Make it work in the editor