diff --git a/spec/hyrax/transactions/steps/ensure_admin_set_spec.rb b/spec/hyrax/transactions/steps/ensure_admin_set_spec.rb
index 798dc161b6..f32d5e70f6 100644
--- a/spec/hyrax/transactions/steps/ensure_admin_set_spec.rb
+++ b/spec/hyrax/transactions/steps/ensure_admin_set_spec.rb
@@ -14,8 +14,8 @@
     end
 
     context 'with an admin set' do
-      let(:admin_set) { create(:admin_set) }
-      let(:work)      { build(:generic_work, admin_set_id: admin_set.id) }
+      let(:admin_set) { valkyrie_create(:hyrax_admin_set) }
+      let(:work)      { build(:hyrax_work, admin_set_id: admin_set.id) }
 
       it 'is a success' do
         expect(step.call(work)).to be_success