diff --git a/vlasov/beamloading.py b/vlasov/beamloading.py
index 67a863cac819a2cbd40d46315f3d2aaf18cf3f79..07688a10ae62b3cf31e6cfe0cef6cdf15f6c930d 100644
--- a/vlasov/beamloading.py
+++ b/vlasov/beamloading.py
@@ -181,7 +181,7 @@ class BeamLoadingVlasov():
         Vtot = 0
         for i in range(self.n_cavity):
             cavity = self.cavity_list[i]
-            Vtot += cavity.VRF(z, self.F[i], self.PHI[i])
+            Vtot += cavity.VRF(z, self.I0, self.F[i], self.PHI[i])
         return Vtot
     
     def dV(self, z):
@@ -189,7 +189,7 @@ class BeamLoadingVlasov():
         Vtot = 0
         for i in range(self.n_cavity):
             cavity = self.cavity_list[i]
-            Vtot += cavity.dVRF(z, self.F[i], self.PHI[i])
+            Vtot += cavity.dVRF(z, self.I0, self.F[i], self.PHI[i])
         return Vtot
     
     def ddV(self, z):
@@ -197,7 +197,7 @@ class BeamLoadingVlasov():
         Vtot = 0
         for i in range(self.n_cavity):
             cavity = self.cavity_list[i]
-            Vtot += cavity.ddVRF(z, self.F[i], self.PHI[i])
+            Vtot += cavity.ddVRF(z, self.I0, self.F[i], self.PHI[i])
         return Vtot
     
     def deltaVRF(self, z):
@@ -205,7 +205,7 @@ class BeamLoadingVlasov():
         Vtot = 0
         for i in range(self.n_cavity):
             cavity = self.cavity_list[i]
-            Vtot += cavity.deltaVRF(z, self.F[i], self.PHI[i])
+            Vtot += cavity.deltaVRF(z, self.I0, self.F[i], self.PHI[i])
         return Vtot
     
     def plot_dV(self, z1=None, z2=None):