diff --git a/neural_network_example.py b/neural_network_example.py
index 2cbb9f5..21e255c 100644
--- a/neural_network_example.py
+++ b/neural_network_example.py
@@ -19,7 +19,7 @@ class TinyModel(nn.Module):
         x = self.layer2(x)
         return x
 
-    model = TinyModel()
+model = TinyModel()
 loss_fn = nn.MSELoss()
 optimizer = optim.SGD(model.parameters(), lr=0.01)