Revert "Fix a bug in TreeNode deserializer"

This reverts commit 44c073c0c37962e784619766d2dcac3ba9483572.