diff --git a/readme.md b/readme.md
index 77abd43..387902a 100644
--- a/readme.md
+++ b/readme.md
@@ -19,7 +19,7 @@ This repository also includes a pytorch implementation that has been tweaked to
   - Time features: time of day, day of week, month of year, etc
   - Bank holidays
   - Position in sequence: days since start of window
-- Target is: mean power usage on block
+- Target is: mean power usage on block 0
 
 ## Example outputs
 
diff --git a/src/models/modules.py b/src/models/modules.py
index 93237e6..1d6f556 100644
--- a/src/models/modules.py
+++ b/src/models/modules.py
@@ -253,7 +253,7 @@ class Decoder(nn.Module):
         x = self._target_transform(target_x)
 
         # concatenate target_x and representation
-        if r is not None:
+        if self.use_deterministic_path:
             z = torch.cat([r, z], dim=-1)
         representation = torch.cat([z, x], dim=-1)