diff --git a/wbia_miew_id/train.py b/wbia_miew_id/train.py
index 750c06e..bfd6cbc 100644
--- a/wbia_miew_id/train.py
+++ b/wbia_miew_id/train.py
@@ -157,7 +157,7 @@ def run(self):
         else:
             margins = None
 
-        model = MiewIdNet(**dict(config.model_params), margins=margins)
+        model = MiewIdNet(**dict(config.model_params))
         model.to(device)
 
         loss_fn = fetch_loss()