Small UI adjustment