diff --git a/examples/estimation_examples/pzflow_demo.ipynb b/examples/estimation_examples/pzflow_demo.ipynb
index 452ec7e..f6d928a 100644
--- a/examples/estimation_examples/pzflow_demo.ipynb
+++ b/examples/estimation_examples/pzflow_demo.ipynb
@@ -29,7 +29,7 @@
     "import qp\n",
     "import tables_io\n",
     "\n",
-    "from rail.estimation.algos.pzflow_nf_default import PZFlowInformer, PZFlowEstimator\n"
+    "from rail.estimation.algos.pzflow_nf import PZFlowInformer, PZFlowEstimator\n"
    ]
   },
   {