fixed small bug with classes