Fix buttons in editor #7