Small UI improvement