diff --git a/test/io/test_files/templates/partial.dart b/test/io/test_files/templates/partial.dart
index 0c3e1f05f..6ab9f0cfe 100644
--- a/test/io/test_files/templates/partial.dart
+++ b/test/io/test_files/templates/partial.dart
@@ -1,4 +1,4 @@
-part of 'main.dart';
+part of test_files.main;
 
 @NgComponent(
     selector: 'my-component',