diff --git a/init.go b/init.go
index a086b8d..e37ba96 100644
--- a/init.go
+++ b/init.go
@@ -1,7 +1,6 @@
 package apidApigeeSync
 
 import (
-	"database/sql"
 	"fmt"
 	"github.com/30x/apid"
 )
@@ -89,7 +88,7 @@
 	return nil
 }
 
-func createTables(db *sql.DB) {
+func createTables(db apid.DB) {
 	_, err := db.Exec(`
 CREATE TABLE apid_config (
     id text,