diff --git a/mattport/nerf/graph.py b/mattport/nerf/graph.py
index 84b8997ee..968522996 100644
--- a/mattport/nerf/graph.py
+++ b/mattport/nerf/graph.py
@@ -15,7 +15,7 @@ class GraphInputs:
 
 
 class Graph(nn.Module):
-    """_summary_ """
+    """_summary_"""
 
     def __init__(self, modules_config: dict) -> None:
         super().__init__()
@@ -39,7 +39,7 @@ class Graph(nn.Module):
         Returns:
             list: ordering of the module names that should be executed
         """
-        raise NotImplementedError
+        return modules_config.keys()
 
     def forward(self, x):
         """_summary_
diff --git a/mattport/nerf/modules/encoding.py b/mattport/nerf/modules/encoding.py
index 5abf0dbcf..b0985ecfc 100644
--- a/mattport/nerf/modules/encoding.py
+++ b/mattport/nerf/modules/encoding.py
@@ -12,8 +12,6 @@ class Encoding(nn.Module):
         super().__init__()
         self.out_dim = out_dim
 
-        raise NotImplementedError
-
     def encode(self, in_tensor: TensorType[..., "input_dim"]) -> TensorType[..., "output_dim"]:
         """Encodes an input tensor.