diff --git a/README.md b/README.md
index 76b81329..6c000cd0 100644
--- a/README.md
+++ b/README.md
@@ -63,7 +63,7 @@ trainer.fit(model)
 #### Define the model   
 
 ```python
-import torch.nn as nn
+from torch import nn
 
 class My_Model(RootModule):
     def __init__(self):