1 files changed, 2 insertions, 2 deletions
diff --git a/spec/factories/ml/model_versions.rb b/spec/factories/ml/model_versions.rb
index 5ae0446b78d..456d1b1e913 100644
--- a/spec/factories/ml/model_versions.rb
+++ b/spec/factories/ml/model_versions.rb
@@ -2,14 +2,14 @@
 
 FactoryBot.define do
   factory :ml_model_versions, class: '::Ml::ModelVersion' do
-    sequence(:version) { |n| "version#{n}" }
+    sequence(:version) { |n| "1.0.#{n}-alpha+test" }
 
     model { association :ml_models }
     project { model.project }
 
     trait :with_package do
       package do
-        association :ml_model_package, name: model.name, version: version, project_id: project.id
+        association :ml_model_package, name: model.name, version: version, project: project
       end
     end
   end