fixed similar class problem