Decorator on first node after root not working

You could argue that it makes sense for the main tree, but the bug still occurs in subtrees, and there failing at the root is very meaningful - it returns control to the parent tree in the correct spot.