diff --git a/model/modules.py b/model/modules.py
index f694d56a6e..01450bbabc 100644
--- a/model/modules.py
+++ b/model/modules.py
@@ -143,7 +143,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