Fixed bug in showing class