Fix to properly set the root node/