diff --git a/main_test.go b/main_test.go
index 9a3403e2..2499344d 100644
--- a/main_test.go
+++ b/main_test.go
@@ -11,6 +11,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/erikstmartin/go-testdb"
 	"github.com/jinzhu/gorm"
 	_ "github.com/jinzhu/gorm/dialects/mssql"
 	_ "github.com/jinzhu/gorm/dialects/mysql"