diff --git a/norch/nn/module.py b/norch/nn/module.py
index eaabd9a..cebd86f 100644
--- a/norch/nn/module.py
+++ b/norch/nn/module.py
@@ -53,8 +53,6 @@ class Module(ABC):
     def to(self, device):
         for _, _, parameter in self.parameters():
             parameter.to(device)
-
-        return self
     
     def state_dict(self):
         state = OrderedDict()
diff --git a/tests/test_autograd.py b/tests/test_autograd.py
index 82e6b61..6793816 100644
--- a/tests/test_autograd.py
+++ b/tests/test_autograd.py
@@ -9,7 +9,7 @@ class TestTensorAutograd(unittest.TestCase):
         self.device = os.environ.get('device')
         if self.device is None or self.device != 'cuda':
             self.device = 'cpu'
-        
+
         print(f"Running tests on: {self.device}")