Fix editor bug