1 files changed, 3 insertions, 6 deletions
diff --git a/app/finders/projects/ml/model_finder.rb b/app/finders/projects/ml/model_finder.rb
index 9ef5dacb551..99c66f53de7 100644
--- a/app/finders/projects/ml/model_finder.rb
+++ b/app/finders/projects/ml/model_finder.rb
@@ -8,12 +8,9 @@ module Projects
       end
 
       def execute
-        @project
-          .packages
-          .installable
-          .ml_model
-          .order_name_desc_version_desc
-          .select_only_first_by_name
+        ::Ml::Model
+          .by_project(@project)
+          .including_latest_version
           .limit(100) # This is a temporary limit before we add pagination
       end
     end