Fixing issue with classes