Small UI tweak