diff --git a/code/TestSessionEnvironment.php b/code/TestSessionEnvironment.php
index c873784..17da234 100644
--- a/code/TestSessionEnvironment.php
+++ b/code/TestSessionEnvironment.php
@@ -185,7 +185,8 @@ class TestSessionEnvironment extends Object {
 			$conn = DB::getConn();
 
 			if(!$conn) {
-				$conn = DB::connect($databaseConfig);
+				DB::connect($databaseConfig);
+				$conn = DB::getConn();
 			}
 
 			$conn->selectDatabase($database);