Fix button visibility in editor