diff --git a/models/tts/fastspeech2/fs2.py b/models/tts/fastspeech2/fs2.py
index 464179f7..61cab7eb 100644
--- a/models/tts/fastspeech2/fs2.py
+++ b/models/tts/fastspeech2/fs2.py
@@ -215,7 +215,7 @@ def forward(
             x = x + pitch_embedding
         if self.energy_feature_level == "frame_level":
             energy_prediction, energy_embedding = self.get_energy_embedding(
-                x, energy_target, mel_mask, p_control
+                x, energy_target, mel_mask, e_control
             )
             x = x + energy_embedding