diff --git a/notebooks/flow_calibration.ipynb b/notebooks/flow_calibration.ipynb
index 03e78e6..86b6e80 100644
--- a/notebooks/flow_calibration.ipynb
+++ b/notebooks/flow_calibration.ipynb
@@ -7,24 +7,19 @@
    "outputs": [],
    "source": [
     "import numpy as np\n",
-    "\n",
     "import matplotlib.pyplot as plt\n",
-    "from h5py import File\n",
-    "%matplotlib inline\n",
-    "\n",
     "from jax import numpy as jnp\n",
     "import jax\n",
     "from tqdm import tqdm\n",
-    "\n",
-    "from scipy.optimize import fmin_powell\n",
-    "from numpyro.infer import MCMC, NUTS, util, init_to_median\n",
+    "from h5py import File\n",
+    "from numpyro.infer import MCMC, NUTS, init_to_median\n",
+    "import corner\n",
     "\n",
     "import csiborgtools\n",
-    "import corner\n",
     "\n",
     "%load_ext autoreload\n",
     "%autoreload 2\n",
-    "\n",
+    "%matplotlib inline\n",
     "\n",
     "paths = csiborgtools.read.Paths(**csiborgtools.paths_glamdring)"
    ]