Fixed broken add-on due to site changes