diff --git a/semantics/saving.py b/semantics/saving.py
index 653f901..27d981b 100644
--- a/semantics/saving.py
+++ b/semantics/saving.py
@@ -12,16 +12,17 @@ from wordnet.models import Hypernymy, LexicalUnit, Synonymy, Synset
 
 def modify_frames(lemma_id, operations, user):
     store_old_versions(lemma_id, operations, user)
-    make_operations(operations)
+    make_operations(lemma_id, operations)
 
-def make_operations(operations):
+def make_operations(lemma_id, operations):
     translation = {'unit_id': {}, 'frame_id': {}, 'complement_id': {}, 'preference_id': {}}
+    entry = Lemma.objects.get(id=lemma_id).entry_obj
     for operation in operations:
         if operation['operation'] == "create_frame":
             luids = [int(m['id']) for m in operation['meanings']]
             translation['frame_id'][int(operation['id'])] = create_frame(luids)
         elif operation['operation'] == "add_unit":
-            translation['unit_id'][int(operation['unit']['id'])] = add_unit(operation['unit'])
+            translation['unit_id'][int(operation['unit']['id'])] = add_unit(entry, operation['unit'])
         elif operation['operation'] == "remove_frame":
             if int(operation['id']) in translation['frame_id']:
                 frame_id = translation['frame_id'][int(operation['id'])]