Update test_unit.rb