diff --git a/retinal_rl/models/neural_circuit.py b/retinal_rl/models/neural_circuit.py
index 10053bf..c8f5d85 100644
--- a/retinal_rl/models/neural_circuit.py
+++ b/retinal_rl/models/neural_circuit.py
@@ -5,7 +5,7 @@
 from typing import Any, List, Type, get_type_hints
 
 import torch
-import torch.nn as nn
+from torch import nn
 from torchinfo import summary
 
 from retinal_rl.util import Activation
@@ -64,7 +64,6 @@ def scan(self) -> None:
     @abstractmethod
     def forward(self, x: torch.Tensor) -> torch.Tensor:
         """Forward pass of the neural circuit."""
-        x = x
         raise NotImplementedError("Each subclass must implement its own forward method.")
 
     @property