Ui Bug Fixes