diff --git a/inference.ipynb b/inference.ipynb
index 2e94ff3..2aa9b50 100644
--- a/inference.ipynb
+++ b/inference.ipynb
@@ -101,7 +101,7 @@
     "try:\n",
     "    model = model.module\n",
     "except:\n",
-    "    pass\n"
+    "    pass\n",
     "model.load_state_dict({k.replace('module.',''):v for k,v in torch.load(checkpoint_path)['state_dict'].items()})\n",
     "_ = model.eval()"
    ]