lib/puppet-lint/plugins/numeric_variable.rb in puppet-lint-numericvariable-1.0.1 vs lib/puppet-lint/plugins/numeric_variable.rb in puppet-lint-numericvariable-1.0.2

@@ -1,8 +1,9 @@
 PuppetLint.new_check(:numeric_variable) do
+  NUMERIC_VAR_TYPES = Set[:VARIABLE, :UNENC_VARIABLE]
   def check
     tokens.select { |r|
-      VARIABLE_TYPES.include? r.type
+      NUMERIC_VAR_TYPES.include? r.type
     }.each do |token|
       if token.value.match(/^[0-9]+$/)
         notify :warning, {
           :message => 'variable is solely numeric',
           :line    => token.line,