more fix on UI to support new API