Small UI fix