diff --git a/test/integration/client/heroku-pgpass-tests.js b/test/integration/client/heroku-pgpass-tests.js
index c037923..c7aeab9 100644
--- a/test/integration/client/heroku-pgpass-tests.js
+++ b/test/integration/client/heroku-pgpass-tests.js
@@ -22,15 +22,18 @@ var config = {
   ssl: true
 };
 
-// connect & disconnect from heroku
-pg.connect(config, assert.success(function(client, done) {
-  client.query('SELECT NOW() as time', assert.success(function(res) {
-    assert(res.rows[0].time.getTime());
+test('uses password file when PGPASSFILE env variable is set', function() {
+  // connect & disconnect from heroku
+  pg.connect(config, assert.calls(function(err, client, done) {
+    assert.isNull(err);
+    client.query('SELECT NOW() as time', assert.success(function(res) {
+      assert(res.rows[0].time.getTime());
 
-    // cleanup ... remove the env variable
-    delete process.env.PGPASSFILE;
+      // cleanup ... remove the env variable
+      delete process.env.PGPASSFILE;
 
-    done();
-    pg.end();
-  }))
-}));
+      done();
+      pg.end();
+    }))
+  }, 15000));
+});
diff --git a/test/integration/client/heroku-ssl-tests.js b/test/integration/client/heroku-ssl-tests.js
index 5b6b87d..e012a09 100644
--- a/test/integration/client/heroku-ssl-tests.js
+++ b/test/integration/client/heroku-ssl-tests.js
@@ -15,11 +15,14 @@ var config = {
   ssl: true
 };
 
-//connect & disconnect from heroku
-pg.connect(config, assert.success(function(client, done) {
-  client.query('SELECT NOW() as time', assert.success(function(res) {
-    assert(res.rows[0].time.getTime());
-    done();
-    pg.end();
-  }))
-}));
+test('connection with config ssl = true', function() {
+  //connect & disconnect from heroku
+  pg.connect(config, assert.calls(function(err, client, done) {
+    assert.isNull(err);
+    client.query('SELECT NOW() as time', assert.success(function(res) {
+      assert(res.rows[0].time.getTime());
+      done();
+      pg.end();
+    }))
+  }, 15000));
+});
\ No newline at end of file
diff --git a/test/test-helper.js b/test/test-helper.js
index 757b200..0a1e94c 100644
--- a/test/test-helper.js
+++ b/test/test-helper.js
@@ -133,9 +133,10 @@ assert.lengthIs = function(actual, expectedLength) {
 
 var expect = function(callback, timeout) {
   var executed = false;
+  timeout = timeout || 5000;
   var id = setTimeout(function() {
-    assert.ok(executed, "Expected execution of function to be fired");
-  }, timeout || 5000)
+    assert.ok(executed, "Expected execution of function to be fired within " + timeout + ' milliseconds');
+  }, timeout)
 
   if(callback.length < 3) {
     return function(err, queryResult) {