diff --git a/backend/memory_management.py b/backend/memory_management.py
index 86e9f09f..4f64c61a 100644
--- a/backend/memory_management.py
+++ b/backend/memory_management.py
@@ -375,7 +375,7 @@ class LoadedModel:
         self.model.model_patches_to(self.model.model_dtype())
 
         try:
-            self.real_model = self.model.forge_patch_model(device_to=patch_model_to)
+            self.real_model = self.model.forge_patch_model(patch_model_to)
         except Exception as e:
             self.model.forge_unpatch_model(self.model.offload_device)
             self.model_unload()