diff --git a/medcat-trainer/webapp/api/api/models.py b/medcat-trainer/webapp/api/api/models.py
index 08ea6c877..eee883c31 100644
--- a/medcat-trainer/webapp/api/api/models.py
+++ b/medcat-trainer/webapp/api/api/models.py
@@ -137,7 +137,7 @@ def save(self, *args, skip_load=False, **kwargs):
         else:
             # For new objects, just update the FK fields without full save
             # Fixes psycopg.errors.UniqueViolation: duplicate key value violates unique constraint "api_modelpack_pkey" 
-            super().save(update_fields=['concept_db', 'vocab'])
+            super().save(*args, update_fields=['concept_db', 'vocab'], **kwargs)
 
     def __str__(self):
         return self.name