fix for tree json notebook