Removed class name bug