diff --git a/server/routes/auth.js b/server/routes/auth.js
index 701b4e1..898f3c3 100644
--- a/server/routes/auth.js
+++ b/server/routes/auth.js
@@ -64,6 +64,12 @@ const multiAuth = new MultiTenantAuth();
 // Initialize IP restriction middleware
 const ipRestriction = new IPRestrictionMiddleware();
 
+// Set test models if available (for testing)
+if (global.__TEST_MODELS__) {
+  multiAuth.setModels && multiAuth.setModels(global.__TEST_MODELS__);
+  ipRestriction.setModels(global.__TEST_MODELS__);
+}
+
 // Session middleware for OAuth state management
 router.use(session({
   secret: process.env.SESSION_SECRET || 'your-session-secret',