fixed class name issue