Improvements to RegisterOnUpdateEditor