diff --git a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb
index 746e6b282..7d92e6be1 100644
--- a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb
+++ b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb
@@ -189,6 +189,7 @@
    },
    "outputs": [],
    "source": [
+    "from mct_quantizers import QuantizationMethod\n",
     "from model_compression_toolkit.constants import FLOAT_BITWIDTH\n",
     "from model_compression_toolkit.target_platform_capabilities.constants import KERNEL_ATTR, BIAS_ATTR\n",
     "from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import schema\n",