diff --git a/migration.go b/migration.go
index 72a0e99..dcb9db2 100644
--- a/migration.go
+++ b/migration.go
@@ -4,6 +4,7 @@ import (
 	"database/sql"
 	"fmt"
 	"git.selly.red/Selly-Modules/logger"
+	"github.com/golang-migrate/migrate/v4"
 	"github.com/golang-migrate/migrate/v4/database/postgres"
 	_ "github.com/golang-migrate/migrate/v4/source/file"
 	"os"