Fix linear classifier