Fixed bad output class setting