diff --git a/examples/pytorch_sparse_quantized_transfer_learning/pytorch_sparse_quantized_transfer_learning.ipynb b/examples/pytorch_sparse_quantized_transfer_learning/pytorch_sparse_quantized_transfer_learning.ipynb
index b46729de0ce..63002433da0 100644
--- a/examples/pytorch_sparse_quantized_transfer_learning/pytorch_sparse_quantized_transfer_learning.ipynb
+++ b/examples/pytorch_sparse_quantized_transfer_learning/pytorch_sparse_quantized_transfer_learning.ipynb
@@ -332,13 +332,13 @@
     "from sparseml.pytorch.utils import ModuleExporter\n",
     "\n",
     "save_dir = \"pytorch_sparse_quantized_transfer_learning\"\n",
-    "qat_onnx_graph_name = \"resnet50_imagenette_pruned_qat.onnx\"\n",
-    "quantized_onnx_path = os.path.join(save_dir, \"resnet50_imagenette_pruned_quant.onnx\")\n",
+    "quant_onnx_graph_name = \"resnet50_imagenette_pruned_quant.onnx\"\n",
+    "quantized_onnx_path = os.path.join(save_dir, quant_onnx_graph_name)\n",
     "\n",
     "exporter = ModuleExporter(model, output_dir=save_dir)\n",
     "exporter.export_pytorch(name=\"resnet50_imagenette_pruned_qat.pth\")\n",
     "exporter.export_onnx(\n",
-    "    torch.randn(1, 3, 224, 224), name=qat_onnx_graph_name, convert_qat=True\n",
+    "    torch.randn(1, 3, 224, 224), name=quant_onnx_graph_name, convert_qat=True\n",
     ")\n",
     "\n",
     "print(f\"Sparse-Quantized ONNX model saved to {quantized_onnx_path}\")"