diff --git a/freeforestml/tests/test_model.py b/freeforestml/tests/test_model.py
index d7c9b809bd5fd749bedd6d0a1c92e24f172ced70..a00a65d412097d6a5aa2fad05c28277980d68587 100644
--- a/freeforestml/tests/test_model.py
+++ b/freeforestml/tests/test_model.py
@@ -909,7 +909,7 @@ class HepNetTestCase(unittest.TestCase):
         df["is_ztt"] = (df.fpid == 0)
 
         net.fit(df.compute(), epochs=5, verbose=0, 
-                weight=Variable("weight", "weight"))
+                event_weight=Variable("weight", "weight"))
 
         fd, path = tempfile.mkstemp()