diff --git a/client/client_test.go b/client/client_test.go
index 834e106..f3716f6 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -372,7 +372,7 @@ func TestUpdate(t *testing.T) {
 
 	// Then check the status command again to confirm the update worked
 	out = RunInteractiveBashCommands(t, `hishtory status`)
-	if !strings.HasPrefix(out, fmt.Sprintf("Hishtory: v0.Unknown\nEnabled: true\nSecret Key: %s\nCommit Hash: ", userSecret)) {
+	if !strings.Contains(out, fmt.Sprintf("\nEnabled: true\nSecret Key: %s\nCommit Hash: ", userSecret)) {
 		t.Fatalf("status command has unexpected output: %#v", out)
 	}
 	if strings.Contains(out, "\nCommit Hash: Unknown\n") {