Update for new addon