small fix for classificatiion