diff --git a/raremodel-tf.py b/raremodel-tf.py
index 42bda47..5bc1c90 100644
--- a/raremodel-tf.py
+++ b/raremodel-tf.py
@@ -315,7 +315,7 @@
 # total_pdf = total_func.as_pdf()
 print(total_pdf.obs)
 
-nll = zfit.loss.UnbinnedNLL(model=total_pdf, data=data, fit_range = (x_min, x_max))
+nll = zfit.loss.UnbinnedNLL(model=total_f, data=data, fit_range = (x_min, x_max))
 
 minimizer = zfit.minimize.MinuitMinimizer()
 result = minimizer.minimize(nll)