fixing test.rb