fix small "UI" issues