Fix classification tests using factories