Fixed class name issue