diff --git a/cmd/callgraph/main_test.go b/cmd/callgraph/main_test.go
index 81fa490c..93ccb618 100644
--- a/cmd/callgraph/main_test.go
+++ b/cmd/callgraph/main_test.go
@@ -40,13 +40,13 @@ func TestCallgraph(t *testing.T) {
 		// tests: main is not called.
 		{"rta", format, true, []string{
 			`pkg.Example --> (pkg.C).f`,
-			`test$main.init --> pkg.init`,
+			`testmain.init --> pkg.init`,
 		}},
 		{"pta", format, true, []string{
 			`<root> --> pkg.Example`,
-			`<root> --> test$main.init`,
+			`<root> --> testmain.init`,
 			`pkg.Example --> (pkg.C).f`,
-			`test$main.init --> pkg.init`,
+			`testmain.init --> pkg.init`,
 		}},
 	} {
 		stdout = new(bytes.Buffer)