* Small UI fix