1 files changed, 6 insertions, 11 deletions
diff --git a/spec/factories/ml/candidates.rb b/spec/factories/ml/candidates.rb
index 9bfb78066bd..27b31524b1f 100644
--- a/spec/factories/ml/candidates.rb
+++ b/spec/factories/ml/candidates.rb
@@ -1,10 +1,9 @@
 # frozen_string_literal: true
 FactoryBot.define do
   factory :ml_candidates, class: '::Ml::Candidate' do
-    association :project, factory: :project
-    association :user
-
-    experiment { association :ml_experiments, project_id: project.id }
+    project { association :project }
+    user { project.owner }
+    experiment { association :ml_experiments, project_id: project.id, user: project.owner }
 
     trait :with_metrics_and_params do
       metrics { Array.new(2) { association(:ml_candidate_metrics, candidate: instance) } }
@@ -16,13 +15,9 @@ FactoryBot.define do
     end
 
     trait :with_artifact do
-      after(:create) do |candidate|
-        candidate.package = FactoryBot.create(
-          :generic_package,
-          name: candidate.package_name,
-          version: candidate.package_version,
-          project: candidate.project
-        )
+      artifact do
+        association(:generic_package, name: instance.package_name, version: instance.package_version || '1',
+          project: project)
       end
     end
   end