Fixed returning visuals to right state

Fixed returning visuals to right state