diff --git a/mattport/nerf/modules/__init__.py b/mattport/nerf/modules/__init__.py
index d59de93cb..39b25225c 100644
--- a/mattport/nerf/modules/__init__.py
+++ b/mattport/nerf/modules/__init__.py
@@ -1,2 +1,2 @@
 from .encoding import Encoding
-from .mlp import MLP
\ No newline at end of file
+from .mlp import MLP
diff --git a/mattport/nerf/modules/encoding.py b/mattport/nerf/modules/encoding.py
index 821c663c5..5abf0dbcf 100644
--- a/mattport/nerf/modules/encoding.py
+++ b/mattport/nerf/modules/encoding.py
@@ -10,7 +10,9 @@ class Encoding(nn.Module):
 
     def __init__(self, out_dim: int) -> None:
         super().__init__()
-        # raise NotImplementedError
+        self.out_dim = out_dim
+
+        raise NotImplementedError
 
     def encode(self, in_tensor: TensorType[..., "input_dim"]) -> TensorType[..., "output_dim"]:
         """Encodes an input tensor.