Small UI Fix