Fixed losing tree expansion state on save.