diff --git a/package/samplers/hebo_base_sampler/sampler.py b/package/samplers/hebo_base_sampler/sampler.py
index c415261d..0fe55356 100644
--- a/package/samplers/hebo_base_sampler/sampler.py
+++ b/package/samplers/hebo_base_sampler/sampler.py
@@ -182,9 +182,11 @@ def _convert_to_hebo_design_space(
         return DesignSpace().parse(design_space)
 
     def infer_relative_search_space(
-            self, study: Study, trial: FrozenTrial
-        ) -> dict[str, BaseDistribution]:
-        return optuna.search_space.intersection_search_space(study._get_trials(deepcopy=False, use_cache=True))
+        self, study: Study, trial: FrozenTrial
+    ) -> dict[str, BaseDistribution]:
+        return optuna.search_space.intersection_search_space(
+            study._get_trials(deepcopy=False, use_cache=True)
+        )
 
     def sample_independent(
         self,