- Fixed class name issue