Fix to work with newer edit urls as well