fix classification mechanism