diff --git a/docs/source/onnx_dynamo.rst b/docs/source/onnx_dynamo.rst
index 09a09bc3a300a..be6b9d48a8d39 100644
--- a/docs/source/onnx_dynamo.rst
+++ b/docs/source/onnx_dynamo.rst
@@ -53,6 +53,7 @@ See below a demonstration of exporter API in action with a simple Multilayer Per
 .. code-block:: python
 
   import torch
+  import torch.nn as nn
 
   class MLPModel(nn.Module):
     def __init__(self):