Change to test.rb