Fix erro in generator.rb