Fixed bug with classes displaying.