diff --git a/docs/source/notes/extending.rst b/docs/source/notes/extending.rst
index 996a5e8ac6e6..6a915163f293 100644
--- a/docs/source/notes/extending.rst
+++ b/docs/source/notes/extending.rst
@@ -643,8 +643,8 @@ implementation more permissive about what operations are allowed::
       def __torch_function__(cls, func, types, args=(), kwargs=None):
           if kwargs is None:
               kwargs = {}
-          args = [a._t if hasattr(a, '_t') else a for a in args]
           metadatas = tuple(a._metadata for a in args if hasattr(a, '_metadata'))
+          args = [a._t if hasattr(a, '_t') else a for a in args]
           assert len(metadatas) > 0
           ret = func(*args, **kwargs)
           return MetadataTensor(ret, metadata=metadatas[0])