diff --git a/lib/hash_syntax/runner.rb b/lib/hash_syntax/runner.rb index 2ac9ca9..ac44635 100644 --- a/lib/hash_syntax/runner.rb +++ b/lib/hash_syntax/runner.rb @@ -34,10 +34,10 @@ def validate_options(opts) def gather_files if ARGV.empty? - AUTO_SUBDIRS.map { |dir| Dir["#{Dir.pwd}/#{dir}/**/*.rb"] }.flatten + AUTO_SUBDIRS.map { |dir| Dir["#{Dir.pwd}/#{dir}/**/*.{rb,html.haml}"] }.flatten else ARGV end end end -end \ No newline at end of file +end