Fixed broken classification input