Minor UI Fix