diff --git a/src/jimgw/likelihood.py b/src/jimgw/likelihood.py
index de274e66..3a236936 100644
--- a/src/jimgw/likelihood.py
+++ b/src/jimgw/likelihood.py
@@ -180,10 +180,6 @@ def __init__(
 
         print("Finding reference parameters..")
         
-        # Check if shape of bounds is correct, otherwise, transpose bounds
-        if jnp.shape(bounds)[0] != 2:
-            bounds = bounds.T
-
         self.ref_params = self.maximize_likelihood(
             bounds=bounds, prior=prior, popsize=popsize, n_loops=n_loops
         )