Fix tests
NoMethodError: undefined method `assert_raise'