API-54 Fix bugs