diff --git a/test/io/source_metadata_extractor_spec.dart b/test/io/source_metadata_extractor_spec.dart
index c332e90e8..cbf19d355 100644
--- a/test/io/source_metadata_extractor_spec.dart
+++ b/test/io/source_metadata_extractor_spec.dart
@@ -15,7 +15,7 @@ void main() {
       sourceMetadataExtractor
       .gatherDirectiveInfo('test/io/test_files/main.dart', sourceCrawler);
 
-      expect(directives, hasLength(4));
+      expect(directives.map((d) => d.selector), unorderedEquals(['[ng-if]', 'my-component']));
 
       DirectiveInfo info = directives.elementAt(1);
       expect(info.expressionAttrs, unorderedEquals(['expr', 'another-expression',
diff --git a/test/io/test_files/main.dart b/test/io/test_files/main.dart
index 508fb45a9..6b7c9e4a1 100644
--- a/test/io/test_files/main.dart
+++ b/test/io/test_files/main.dart
@@ -1,7 +1,5 @@
 library test_files.main;
 
-import 'package:angular/core/module.dart';
-
 @NgDirective(
     children: NgAnnotation.TRANSCLUDE_CHILDREN,
     selector:'[ng-if]',