1 files changed, 2 insertions, 3 deletions

diff --git a/cmd/register/register.go b/cmd/register/register.go
index fdd083c..c00b54b 100644
--- a/cmd/register/register.go
+++ b/cmd/register/register.go
@@ -10,7 +10,6 @@ import (
         glecho "code.crute.us/mcrute/golib/echo"
         "code.crute.us/mcrute/ssh-proxy/app"
         "code.crute.us/mcrute/ssh-proxy/app/models"
-        "code.crute.us/mcrute/ssh-proxy/db"
         "github.com/spf13/cobra"
 )
 
@@ -43,13 +42,13 @@ func registerMain(cfg app.Config, username string) {
                 log.Fatalf("Error connecting to mongodb: %s", err)
         }
 
-        userStore := &db.MongoDbBasicStore[*models.User]{
+        userStore := &mongodb.MongoDbBasicStore[*models.User]{
                 Db:             mongo,
                 CollectionName: "users",
         }
 
         authSessionStore := &models.AuthSessionStoreMongodb{
-                MongoDbBasicStore: &db.MongoDbBasicStore[*models.AuthSession]{
+                MongoDbBasicStore: &mongodb.MongoDbBasicStore[*models.AuthSession]{
                         Db:             mongo,
                         CollectionName: "oauth_sessions",
                 },