diff --git a/hyperlight/hypernet/model.py b/hyperlight/hypernet/model.py
index e352c48..07be91c 100644
--- a/hyperlight/hypernet/model.py
+++ b/hyperlight/hypernet/model.py
@@ -106,12 +106,12 @@ def reset_parameters(self) -> None:
                     nonlinearity=self.nonlinearity,
                 )
 
-            initialize_layer(
-                self.output,
-                self.init_distribution,
-                init_bias=self.init_bias,
-                nonlinearity=None,
-            )
+        initialize_layer(
+            self.output,
+            self.init_distribution,
+            init_bias=self.init_bias,
+            nonlinearity=None,
+        )
 
 
 # TODO support for categoricals via 'C' shape and frozen embeddings