fixed image class problem