diff --git a/dreamer4/dreamer4.py b/dreamer4/dreamer4.py
index 36f61bf..e83eb97 100644
--- a/dreamer4/dreamer4.py
+++ b/dreamer4/dreamer4.py
@@ -1873,6 +1873,8 @@ class VideoTokenizer(Module):
         if self.has_lpips_loss:
             lpips_loss = self.lpips(video, recon_video)
 
+        decorr_loss = self.zero
+
         if self.encoder_add_decor_aux_loss:
             decorr_loss = self.decorr_loss(time_attn_normed_inputs)
 
diff --git a/pyproject.toml b/pyproject.toml
index 403ef23..fc3047b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "dreamer4"
-version = "0.1.7"
+version = "0.1.8"
 description = "Dreamer 4"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }