diff --git a/gorm_expect.go b/gorm_expect.go
index 4c29231f..df714201 100644
--- a/gorm_expect.go
+++ b/gorm_expect.go
@@ -11,9 +11,15 @@ import (
 type TestHelper struct {
 	gormDb   *DB
 	mockDb   *sql.DB
+	adapter  Adapter
 	asserter Asserter
 }
 
+// Close closes the DB connection
+func (h *TestHelper) Close() error {
+	return h.adapter.Close()
+}
+
 func (h *TestHelper) ExpectFirst(model interface{}) Query {
 	fmt.Printf("Expecting query: %s", "some query\n")
 	return h.asserter.Query("some sql")
@@ -104,5 +110,5 @@ func NewTestHelper(adapter Adapter) (error, *DB, *TestHelper) {
 		return err, nil, nil
 	}
 
-	return nil, gormDb, &TestHelper{gormDb: gormDb, mockDb: mockDb, asserter: asserter}
+	return nil, gormDb, &TestHelper{gormDb: gormDb, mockDb: mockDb, adapter: adapter, asserter: asserter}
 }