1 files changed, 7 insertions, 6 deletions
diff --git a/spec/migrations/cleanup_vulnerability_state_transitions_with_same_from_state_to_state_spec.rb b/spec/migrations/cleanup_vulnerability_state_transitions_with_same_from_state_to_state_spec.rb
index 92ece81ffc8..b808f03428d 100644
--- a/spec/migrations/cleanup_vulnerability_state_transitions_with_same_from_state_to_state_spec.rb
+++ b/spec/migrations/cleanup_vulnerability_state_transitions_with_same_from_state_to_state_spec.rb
@@ -3,10 +3,11 @@
 require 'spec_helper'
 require_migration!
 
-RSpec.describe CleanupVulnerabilityStateTransitionsWithSameFromStateToState, :migration do
-  let_it_be(:namespace) { table(:namespaces).create!(name: 'namespace', type: 'Group', path: 'namespace') }
-  let_it_be(:user) { table(:users).create!(email: 'author@example.com', username: 'author', projects_limit: 10) }
-  let_it_be(:project) do
+RSpec.describe CleanupVulnerabilityStateTransitionsWithSameFromStateToState, :migration,
+feature_category: :vulnerability_management do
+  let!(:namespace) { table(:namespaces).create!(name: 'namespace', type: 'Group', path: 'namespace') }
+  let!(:user) { table(:users).create!(email: 'author@example.com', username: 'author', projects_limit: 10) }
+  let!(:project) do
     table(:projects).create!(
       path: 'project',
       namespace_id: namespace.id,
@@ -14,7 +15,7 @@ RSpec.describe CleanupVulnerabilityStateTransitionsWithSameFromStateToState, :mi
     )
   end
 
-  let_it_be(:vulnerability) do
+  let!(:vulnerability) do
     table(:vulnerabilities).create!(
       project_id: project.id,
       author_id: user.id,
@@ -25,7 +26,7 @@ RSpec.describe CleanupVulnerabilityStateTransitionsWithSameFromStateToState, :mi
     )
   end
 
-  let_it_be(:state_transitions) { table(:vulnerability_state_transitions) }
+  let!(:state_transitions) { table(:vulnerability_state_transitions) }
 
   let!(:state_transition_with_no_state_change) do
     state_transitions.create!(