diff --git a/notebooks/double_source_plane.ipynb b/notebooks/double_source_plane.ipynb
index 524bbaa..1074429 100644
--- a/notebooks/double_source_plane.ipynb
+++ b/notebooks/double_source_plane.ipynb
@@ -295,8 +295,7 @@
    "source": [
     "mcmc_samples, log_prob = mcmc_sampler.mcmc_emcee(\n",
     "    n_walkers, n_burn, n_run, kwargs_mean_start, kwargs_sigma_start\n",
-    ")\n",
-    "\n"
+    ")"
    ]
   },
   {
diff --git a/test/test_Likelihood/test_hierarchy_likelihood.py b/test/test_Likelihood/test_hierarchy_likelihood.py
index cb0d702..28e4834 100644
--- a/test/test_Likelihood/test_hierarchy_likelihood.py
+++ b/test/test_Likelihood/test_hierarchy_likelihood.py
@@ -318,12 +318,13 @@ def test_lum_dist_likelihood(self):
         magnitude_measured = magnification_model * amp_int
         magnitude_measured_cov = np.diag((magnitude_measured / 10) ** 2)
 
-        kwargs_likelihood = {"amp_measured": magnitude_measured,
+        kwargs_likelihood = {
+            "amp_measured": magnitude_measured,
             "cov_amp_measured": magnitude_measured_cov,
             "magnification_model": magnification_model,
             "cov_magnification_model": magnification_model_cov,
-            "magnitude_zero_point": magnitude_zero_point}
-
+            "magnitude_zero_point": magnitude_zero_point,
+        }
 
         likelihood = LensLikelihood(
             z_lens=z_lens,