Fix js issue in UI