diff --git a/example/GW150914.py b/example/GW150914.py
index 38eae6c4..a9c1c9c8 100644
--- a/example/GW150914.py
+++ b/example/GW150914.py
@@ -83,7 +83,13 @@
         sin_dec_prior,
     ]
 )
-likelihood = TransientLikelihoodFD([H1, L1], waveform=RippleIMRPhenomD(), trigger_time=gps, duration=4, post_trigger_duration=2)
+likelihood = TransientLikelihoodFD(
+    [H1, L1],
+    waveform=RippleIMRPhenomD(),
+    trigger_time=gps,
+    duration=4,
+    post_trigger_duration=2,
+)
 
 
 mass_matrix = jnp.eye(11)
diff --git a/src/jimgw/prior.py b/src/jimgw/prior.py
index 5b6fa21f..8a32e2c3 100644
--- a/src/jimgw/prior.py
+++ b/src/jimgw/prior.py
@@ -193,7 +193,7 @@ def sample(self, rng_key: jax.random.PRNGKey, n_samples: int) -> dict:
 
     def log_prob(self, x: dict) -> Float:
         variable = x[self.naming[0]]
-        return jnp.log(jnp.exp(-variable)/(1 + jnp.exp(-variable))**2) 
+        return jnp.log(jnp.exp(-variable) / (1 + jnp.exp(-variable)) ** 2)
 
 
 class Sphere(Prior):
diff --git a/test/test_prior.py b/test/test_prior.py
index ab99431c..d4c5be59 100644
--- a/test/test_prior.py
+++ b/test/test_prior.py
@@ -1 +1 @@
-from jimgw.prior import Uniform, UUniform, Composite
+from jimgw.prior import Uniform, Unconstrained_Uniform, Composite