diff --git a/bilby/core/result.py b/bilby/core/result.py
index 11ec22a0ebb28f409dd41d038f9b33251ed07325..0555b824866956bd0b8dd1ad4210417d7d5aa458 100644
--- a/bilby/core/result.py
+++ b/bilby/core/result.py
@@ -168,10 +168,7 @@ def rejection_sample(posterior, weights):
 
     """
     keep = weights > np.random.uniform(0, max(weights), weights.shape)
-    if isinstance(posterior, np.ndarray):
-        return posterior[keep]
-    elif isinstance(posterior, pd.DataFrame):
-        return posterior.iloc[keep]
+    return posterior[keep]
 
 
 def reweight(result, label=None, new_likelihood=None, new_prior=None,