Small UI bug fixed