AUI-1541

We remove the attribute on the JSON node, but only set it back for the TreeNode. Simple fix is to reset it on the JSON node as well.