Bug fixe for simple UI.