1 files changed, 16 insertions, 1 deletions
diff --git a/go/internal/command/fallback/fallback_test.go b/go/internal/command/fallback/fallback_test.go
index afd752b..91dcb53 100644
--- a/go/internal/command/fallback/fallback_test.go
+++ b/go/internal/command/fallback/fallback_test.go
@@ -6,6 +6,8 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/require"
+
+	"gitlab.com/gitlab-org/gitlab-shell/go/internal/command/commandargs"
 )
 
 type fakeExec struct {
@@ -18,8 +20,21 @@ type fakeExec struct {
 	Env      []string
 }
 
+type FakeCommandArgs struct {
+	executable commandargs.Executable
+	arguments  []string
+}
+
+func (f *FakeCommandArgs) Executable() commandargs.Executable {
+	return f.executable
+}
+
+func (f *FakeCommandArgs) Arguments() []string {
+	return f.arguments
+}
+
 var (
-	fakeArgs = []string{"./test", "foo", "bar"}
+	fakeArgs = &FakeCommandArgs{executable: commandargs.GitlabShell, arguments: []string{"foo", "bar"}}
 )
 
 func (f *fakeExec) Exec(filename string, args []string, env []string) error {