diff --git a/pg/pg-tests.ts b/pg/pg-tests.ts
index 087c60834d..99d1fac9ba 100644
--- a/pg/pg-tests.ts
+++ b/pg/pg-tests.ts
@@ -12,10 +12,13 @@ pg.connect(conString, (err, client, done) => {
         return console.error("Error fetching client from pool", err);
     }
     client.query("SELECT $1::int AS number", ["1"], (err, result) => {
-        done();
         if (err) {
+            done(err);
             return console.error("Error running query", err);
         }
+        else {
+            done();
+        }
         console.log(result.rows[0]["number"]);
         return null;
     });
@@ -37,4 +40,4 @@ client.connect((err) => {
         return null;
     });
     return null;
-});
\ No newline at end of file
+});