Fix new API error