Fixed bug in editor class