Fixed another thing in the Imgseg Lab

Fixed one more thing