diff --git a/codes/models/gpt_voice/lucidrains_dvae.py b/codes/models/gpt_voice/lucidrains_dvae.py
index 37aa8f47..673bf9eb 100644
--- a/codes/models/gpt_voice/lucidrains_dvae.py
+++ b/codes/models/gpt_voice/lucidrains_dvae.py
@@ -228,7 +228,7 @@ class DiscreteVAE(nn.Module):
                 out = d(out)
         else:
             # This is non-differentiable, but gives a better idea of how the network is actually performing.
-            out = self.decode(codes)
+            out, _ = self.decode(codes)
 
         # reconstruction loss
         recon_loss = self.loss_fn(img, out, reduction='none')