Update tests.rb