Small UI fix

Small viewer UI fix