used $set instead of update

db_script_for_location.py

@@ -17,9 +17,5 @@ for record in records:
 
 for record in data_list:
     query = {"task_id": record['task_id']}
-    task_instance_records = list(mydatabase.task_instances.find(query))
-    for task_instance_record in task_instance_records:
-        if 'task_creation_step' in list(task_instance_record.keys()):
-            task_instance_record['task_creation_step'].update({"Location_1": record['Location_1']})
-        mydatabase.task_instances.update_one(query, task_instance_data)
+    task_instance_records = mydatabase.task_instances.find(query,{"$set":{"task_creation_step.Location_1": record['Location_1']}})