diff --git a/models/db/install/db.go b/models/db/install/db.go
index 104a7a8e39..e64e85cf01 100644
--- a/models/db/install/db.go
+++ b/models/db/install/db.go
@@ -6,12 +6,10 @@ package install
 import (
 	"forgejo.org/models/db"
 	"forgejo.org/modules/setting"
-
-	"xorm.io/xorm"
 )
 
-func getXORMEngine() *xorm.Engine {
-	return db.DefaultContext.(*db.Context).Engine().(*xorm.Engine)
+func getXORMEngine() db.Engine {
+	return db.DefaultContext.(*db.Context).Engine()
 }
 
 // CheckDatabaseConnection checks the database connection