Changed assert_raises for assert_raise_message