diff --git a/tft/module.py b/tft/module.py
index 0140fd4..55afb67 100644
--- a/tft/module.py
+++ b/tft/module.py
@@ -569,9 +569,7 @@ class TFTModel(nn.Module):
         c_c = self.state_c(static_var)
         states = [c_h.unsqueeze(0), c_c.unsqueeze(0)]
 
-        enc_out = self.temporal_encoder(
-            past_selection, tgt_input=None, states=states
-        )
+        enc_out = self.temporal_encoder(past_selection, tgt_input=None, states=states)
         dec_output = self.temporal_decoder(enc_out, static_enrichment)
         params = self.param_proj(dec_output)
 
@@ -580,12 +578,13 @@ class TFTModel(nn.Module):
         next_sample = distr.sample()
         future_samples = [next_sample]
         import pdb
+
         pdb.set_trace()
 
         for k in range(1, self.prediction_length):
             # TODO
-            future_target_proj = pass
-            future_time_feat_proj = pass
+            # future_target_proj = pass
+            # future_time_feat_proj = pass
 
             future_selection, _ = self.future_selection(
                 [future_target_proj, future_time_feat_proj], static_selection