Fix bug in classes UI.