Fix assert_raises