Fix classification mistake