Small UI bugfix