Update ui.js

fix the fix...