diff --git a/Examples/torch/quantization/quant_analyzer.ipynb b/Examples/torch/quantization/quant_analyzer.ipynb
index 0b35d392a1f..816da5922a6 100644
--- a/Examples/torch/quantization/quant_analyzer.ipynb
+++ b/Examples/torch/quantization/quant_analyzer.ipynb
@@ -287,7 +287,7 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "from aimet_torch.quant_analyzer import CallbackFunc\n",
+    "from aimet_torch.common.defs import CallbackFunc\n",
     "\n",
     "forward_pass_callback = CallbackFunc(pass_calibration_data, use_cuda)"
    ],