Update ui.js

bug fixed!?