diff --git a/pkg/integration/tests/custom_commands/menu_from_command.go b/pkg/integration/tests/custom_commands/menu_from_command.go
index f45d820d6..765f7b873 100644
--- a/pkg/integration/tests/custom_commands/menu_from_command.go
+++ b/pkg/integration/tests/custom_commands/menu_from_command.go
@@ -55,9 +55,9 @@ var MenuFromCommand = NewIntegrationTest(NewIntegrationTestArgs{
 
 		assert.InMenu()
 		assert.MatchCurrentViewTitle(Equals("Choose commit message"))
-		assert.MatchSelectedLine(Equals("baz"))
+		assert.MatchSelectedLine(Equals("baz ")) // TODO: remove the trailing space
 		input.NextItem()
-		assert.MatchSelectedLine(Equals("bar"))
+		assert.MatchSelectedLine(Equals("bar ")) // TODO: remove the trailing space
 		input.Confirm()
 
 		assert.InPrompt()