fix ui issues.