diff --git a/spec/services/projects/after_import_service_spec.rb b/spec/services/projects/after_import_service_spec.rb
index 7289edd1e91854210c1ff4d2a380a7435a896deb..4f74a779cd520bfc1b43ec296c04ea413d15a283 100644
--- a/spec/services/projects/after_import_service_spec.rb
+++ b/spec/services/projects/after_import_service_spec.rb
@@ -77,9 +77,8 @@
 
       before do
         call_count = 0
-        original_method = repository.method(:delete_all_refs_except)
 
-        allow(repository).to receive(:delete_all_refs_except) do |*args|
+        allow(repository).to receive(:delete_all_refs_except).and_wrap_original do |original_method, *args|
           call_count += 1
           call_count > 1 ? original_method.call(*args) : raise(exception)
         end