1 files changed, 2 insertions, 4 deletions
diff --git a/qa/qa/page/group/settings/group_deploy_tokens.rb b/qa/qa/page/group/settings/group_deploy_tokens.rb
index 7d908f473de..c1c3303113b 100644
--- a/qa/qa/page/group/settings/group_deploy_tokens.rb
+++ b/qa/qa/page/group/settings/group_deploy_tokens.rb
@@ -54,12 +54,10 @@ module QA
 
           private
 
-          def within_new_project_deploy_token
+          def within_new_project_deploy_token(&block)
             has_element?(:created_deploy_token_container, wait: QA::Support::Repeater::DEFAULT_MAX_WAIT_TIME)
 
-            within_element(:created_deploy_token_container) do
-              yield
-            end
+            within_element(:created_deploy_token_container, &block)
           end
         end
       end