diff --git a/spec/support/test_env.rb b/spec/support/test_env.rb
index be5331e4770a35f205beaa2640f4473e3b5a86d3..9f9ef20f99b55015692d58f5bd0e92b5d53a2ee3 100644
--- a/spec/support/test_env.rb
+++ b/spec/support/test_env.rb
@@ -81,8 +81,7 @@ module TestEnv
 
   def setup_gitlab_shell
     unless File.directory?(Gitlab.config.gitlab_shell.path)
-      # TODO: Remove `[shards]` when gitlab-shell v3.1.0 is published
-      `rake gitlab:shell:install[shards]`
+      `rake gitlab:shell:install`
     end
   end