fix class nonsense