diff --git a/reverse_engineering/api.js b/reverse_engineering/api.js
index fa69ce5..8900c26 100644
--- a/reverse_engineering/api.js
+++ b/reverse_engineering/api.js
@@ -62,7 +62,7 @@ module.exports = {
 		}
 	},
 
-	disconnect: function (connectionInfo, cb, app) {
+	disconnect: function (connectionInfo, logger, cb, app) {
 		cassandraHelper(app.require('lodash')).close(app);
 		cb();
 	},
@@ -80,7 +80,7 @@ module.exports = {
 					logger.log('info', 'Connection successful', 'Test connection');
 				}
 
-				this.disconnect(connectionInfo, () => {}, app);
+				this.disconnect(connectionInfo, logger, () => {}, app);
 
 				return cb(cassandraHelper(app.require('lodash')).prepareError(error));
 			},