fix js with ui change