diff --git a/aiida_common_workflows/cli/utils.py b/aiida_common_workflows/cli/utils.py
index 606eefa6..5d2d168a 100644
--- a/aiida_common_workflows/cli/utils.py
+++ b/aiida_common_workflows/cli/utils.py
@@ -84,7 +84,7 @@ def get_code_from_list_or_database(codes, entry_point: str):
         if entry.default_calc_job_plugin == entry_point:
             return entry
 
-    result = QueryBuilder().append(AbstractCode, filters={'attributes.default_calc_job_plugin': entry_point}).first()
+    result = QueryBuilder().append(AbstractCode, filters={'attributes.input_plugin': entry_point}).first()
 
     if result is not None:
         return result[0]