diff --git a/notebooks/end2end_example/tfc_end2end_example.ipynb b/notebooks/end2end_example/tfc_end2end_example.ipynb
index 7c6c70f246edd5a829938429d01cd8f4ca34c166..b3b409760eda32a7ae055feec3d0b4c4f8edaf23 100644
--- a/notebooks/end2end_example/tfc_end2end_example.ipynb
+++ b/notebooks/end2end_example/tfc_end2end_example.ipynb
@@ -1939,7 +1939,7 @@
     "from finn.core.throughput_test import throughput_test\n",
     "\n",
     "child_model = ModelWrapper(getCustomOp(sdp_node).get_nodeattr(\"model\"))\n",
-    "res = throughput_test(child_model)\n",
+    "res = throughput_test_remote(child_model)\n",
     "print(\"Network metrics:\")\n",
     "for key in res:\n",
     "    print(str(key) + \": \" + str(res[key]))"