diff --git a/cmd/run_test.go b/cmd/run_test.go
index bed9c2a..1bf82cf 100644
--- a/cmd/run_test.go
+++ b/cmd/run_test.go
@@ -334,8 +334,8 @@ func TestRunPRDescribe(t *testing.T) {
 		if mock.calls != 1 {
 			t.Errorf("expected 1 AI call, got %d", mock.calls)
 		}
-		// Expect "diff", "develop..HEAD", "--no-color"
-		expectedArg := "develop..HEAD"
+		// Expect "diff", "origin/develop..HEAD", "--no-color"
+		expectedArg := "origin/develop..HEAD"
 		found := false
 		for _, arg := range capturedArgs {
 			if arg == expectedArg {
@@ -362,7 +362,7 @@ func TestRunPRDescribe(t *testing.T) {
 		if mock.calls != 1 {
 			t.Errorf("expected 1 AI call, got %d", mock.calls)
 		}
-		expectedArg := "master..HEAD"
+		expectedArg := "origin/master..HEAD"
 		found := false
 		for _, arg := range capturedArgs {
 			if arg == expectedArg {