Update scaffold_generator.rb