diff --git a/src/caosadvancedtools/models/data_model.py b/src/caosadvancedtools/models/data_model.py
index ea864bde9fa3859c4853aa1daf40df56505cd769..4d9419936ec2903d519dca5f9373c31d33b26fe1 100644
--- a/src/caosadvancedtools/models/data_model.py
+++ b/src/caosadvancedtools/models/data_model.py
@@ -251,9 +251,7 @@ class DataModel(dict):
 
                 for par in entity.get_parents():
                     if par.name.lower() == valid_e.name.lower():
-                        # ToDo: Fix https://gitlab.indiscale.com/caosdb/src/caosdb-advanced-user-tools/-/issues/140
-                        #       and remove pylint disable, or close and leave
-                        par._wrap(valid_e)              # pylint: disable=protected-access
+                        par.id = valid_e.id
 
     def collect_entities(self):
         """ Collects all entities: explicitly defined RecordTypes and