fix main.rb