diff --git a/encodec/modules/norm.py b/encodec/modules/norm.py
index 19970e0..8ec476d 100644
--- a/encodec/modules/norm.py
+++ b/encodec/modules/norm.py
@@ -25,4 +25,4 @@ def forward(self, x):
         x = einops.rearrange(x, 'b ... t -> b t ...')
         x = super().forward(x)
         x = einops.rearrange(x, 'b t ... -> b ... t')
-        return
+        return x