Update extension code for new API