diff --git a/norch/__pycache__/tensor.cpython-38.pyc b/norch/__pycache__/tensor.cpython-38.pyc
index 445c3e3..882aabd 100644
Binary files a/norch/__pycache__/tensor.cpython-38.pyc and b/norch/__pycache__/tensor.cpython-38.pyc differ
diff --git a/norch/autograd/__pycache__/functions.cpython-38.pyc b/norch/autograd/__pycache__/functions.cpython-38.pyc
index 154463f..11732bb 100644
Binary files a/norch/autograd/__pycache__/functions.cpython-38.pyc and b/norch/autograd/__pycache__/functions.cpython-38.pyc differ
diff --git a/norch/nn/__pycache__/loss.cpython-38.pyc b/norch/nn/__pycache__/loss.cpython-38.pyc
index 52d18ca..26a0e8b 100644
Binary files a/norch/nn/__pycache__/loss.cpython-38.pyc and b/norch/nn/__pycache__/loss.cpython-38.pyc differ
diff --git a/norch/nn/__pycache__/module.cpython-38.pyc b/norch/nn/__pycache__/module.cpython-38.pyc
index 1a7a26c..f03aec3 100644
Binary files a/norch/nn/__pycache__/module.cpython-38.pyc and b/norch/nn/__pycache__/module.cpython-38.pyc differ
diff --git a/norch/nn/functional.py b/norch/nn/functional.py
index 4c4d757..d63e3d2 100644
--- a/norch/nn/functional.py
+++ b/norch/nn/functional.py
@@ -23,10 +23,12 @@ def softmax(x, dim=None):
     
 def one_hot_encode(x, num_classes):
     one_hot = [[0] * num_classes for _ in range(x.numel)]
-    
+
     # Set the appropriate elements to 1
     for i in range(x.numel):
-        target_idx = int(x.tensor.contents.data[i])
+        target_idx = int(x[i])
         one_hot[i][target_idx] = 1
 
+    
+
     return norch.Tensor(one_hot)
\ No newline at end of file