Fixed up issue with class