diff --git a/codes/models/audio/music/tfdpc_v5.py b/codes/models/audio/music/tfdpc_v5.py
index bbe5ee50..a71650ae 100644
--- a/codes/models/audio/music/tfdpc_v5.py
+++ b/codes/models/audio/music/tfdpc_v5.py
@@ -432,7 +432,7 @@ def inference_tfdpc5_with_cheater():
             spectral_diffuser = SpacedDiffusion(use_timesteps=space_timesteps(4000, [32]), model_mean_type='epsilon',
                                    model_var_type='learned_range', loss_type='mse', betas=get_named_beta_schedule('linear', 4000),
                                    conditioning_free=True, conditioning_free_k=1)
-            from trainer.injectors.audio_injectors import
+            from trainer.injectors.audio_injectors import denormalize_torch_mel
             gen_mel_denorm = denormalize_torch_mel(gen_mel)
             output_shape = (1,16,gen_mel_denorm.shape[-1]*256//16)
             gen_wav = spectral_diffuser.ddim_sample_loop(m2w, output_shape, model_kwargs={'codes': gen_mel_denorm})