Remove assert_raises as decorator