diff --git a/lib/src/analyzer.dart b/lib/src/analyzer.dart
index 0bfb2f3..5a2b59e 100644
--- a/lib/src/analyzer.dart
+++ b/lib/src/analyzer.dart
@@ -508,7 +508,7 @@
 class _TopLevelIncludeReplacer extends Visitor {
   final Messages _messages;
   final IncludeDirective _include;
-  final List<RuleSet> _newRules;
+  final List<TreeNode> _newRules;
   bool _foundAndReplaced = false;
 
   /**
@@ -516,7 +516,7 @@
    * with the [newRules].  If [ruleSet] is found and replaced return true.
    */
   static bool replace(Messages messages, StyleSheet styleSheet,
-      IncludeDirective include, List<RuleSet> newRules) {
+      IncludeDirective include, List<TreeNode> newRules) {
     var visitor = new _TopLevelIncludeReplacer(messages, include, newRules);
     visitor.visitStyleSheet(styleSheet);
     return visitor._foundAndReplaced;