diff --git a/pts/modules/flows.py b/pts/modules/flows.py
index b00ab58..52f1e09 100644
--- a/pts/modules/flows.py
+++ b/pts/modules/flows.py
@@ -327,8 +327,6 @@ def forward(self, x, cond):
         if self.scale is not None:
             x /= self.scale
         u, log_abs_det_jacobian = self.net(x, cond)
-        if self.scale is not None:
-            log_abs_det_jacobian -= torch.log(torch.abs(self.scale))
         return u, log_abs_det_jacobian
 
     def inverse(self, u, cond):