diff --git a/xopt/generators/bayesian/expected_improvement.py b/xopt/generators/bayesian/expected_improvement.py
index daeda70d..5db2edb3 100644
--- a/xopt/generators/bayesian/expected_improvement.py
+++ b/xopt/generators/bayesian/expected_improvement.py
@@ -1,6 +1,8 @@
 import torch
 from botorch.acquisition import (
-    ScalarizedPosteriorTransform, LogExpectedImprovement, qLogExpectedImprovement,
+    ScalarizedPosteriorTransform,
+    LogExpectedImprovement,
+    qLogExpectedImprovement,
 )
 
 from xopt.generators.bayesian.bayesian_generator import (