diff --git a/tools/pytorch-quantization/examples/calibrate_quant_resnet50.ipynb b/tools/pytorch-quantization/examples/calibrate_quant_resnet50.ipynb
index bf0c427a..17e4377e 100644
--- a/tools/pytorch-quantization/examples/calibrate_quant_resnet50.ipynb
+++ b/tools/pytorch-quantization/examples/calibrate_quant_resnet50.ipynb
@@ -529,7 +529,7 @@
     "traindir = os.path.join(data_path, 'train')\n",
     "valdir = os.path.join(data_path, 'val')\n",
     "_args = collections.namedtuple('mock_args', ['model', 'distributed', 'cache_dataset'])\n",
-    "dataset, dataset_test, train_sampler, test_sampler = load_data(traindir, valdir, _args(model=model_name, distributed=False, cache_dataset=False))\n",
+    "dataset, dataset_test, train_sampler, test_sampler = load_data(traindir, valdir, _args(model='resnet50', distributed=False, cache_dataset=False))\n",
     "\n",
     "data_loader = torch.utils.data.DataLoader(\n",
     "    dataset, batch_size=batch_size,\n",