Update for new extension