diff --git a/dreamer4/dreamer4.py b/dreamer4/dreamer4.py
index 4d4d7dc..a141007 100644
--- a/dreamer4/dreamer4.py
+++ b/dreamer4/dreamer4.py
@@ -2431,10 +2431,10 @@ class DynamicsWorldModel(Module):
             )
 
             if exists(discrete_log_probs):
-                behavior_clone_loss = behavior_clone_loss + discrete_log_probs.sum(dim = -1).mean()
+                behavior_clone_loss = behavior_clone_loss - discrete_log_probs.sum(dim = -1).mean()
 
             if exists(continuous_log_probs):
-                behavior_clone_loss = behavior_clone_loss + continuous_log_probs.sum(dim = -1).mean()
+                behavior_clone_loss = behavior_clone_loss - continuous_log_probs.sum(dim = -1).mean()
 
         # gather losses
 
diff --git a/pyproject.toml b/pyproject.toml
index d48ab68..af698ba 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "dreamer4"
-version = "0.0.35"
+version = "0.0.36"
 description = "Dreamer 4"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }