diff --git a/scripts/helper.py b/scripts/helper.py
index 64934a66c12e..229c5e116198 100644
--- a/scripts/helper.py
+++ b/scripts/helper.py
@@ -43,7 +43,7 @@
         hda_id = int(args.hda_id)
     except Exception:
         hda_id = int(helper.decode_id(args.hda_id))
-    hda = model.context.current.query(model.HistoryDatasetAssociation).get(hda_id)
+    hda = model.session.get(model.HistoryDatasetAssociation, hda_id)
     print(f'HDA "{hda.id}" is Dataset "{hda.dataset.id}" at: {hda.file_name}')
 
 if args.ldda_id:
@@ -51,5 +51,5 @@
         ldda_id = int(args.ldda_id)
     except Exception:
         ldda_id = int(helper.decode_id(args.ldda_id))
-    ldda = model.context.current.query(model.HistoryDatasetAssociation).get(ldda_id)
+    ldda = model.session.get(model.LibraryDatasetDatasetAssociation, ldda_id)
     print(f'LDDA "{ldda.id}" is Dataset "{ldda.dataset.id}" at: {ldda.file_name}')