diff --git a/conda-envs/mlos-3.10.yml b/conda-envs/mlos-3.10.yml
index 8793b14838..31e313f0bc 100644
--- a/conda-envs/mlos-3.10.yml
+++ b/conda-envs/mlos-3.10.yml
@@ -28,6 +28,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4
diff --git a/conda-envs/mlos-3.11.yml b/conda-envs/mlos-3.11.yml
index 76866df3ab..22a3390e6a 100644
--- a/conda-envs/mlos-3.11.yml
+++ b/conda-envs/mlos-3.11.yml
@@ -28,6 +28,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4
diff --git a/conda-envs/mlos-3.12.yml b/conda-envs/mlos-3.12.yml
index 4b61bb750a..18dc02293f 100644
--- a/conda-envs/mlos-3.12.yml
+++ b/conda-envs/mlos-3.12.yml
@@ -30,6 +30,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4
diff --git a/conda-envs/mlos-3.13.yml b/conda-envs/mlos-3.13.yml
index fc32e92800..5ff0e8e5a3 100644
--- a/conda-envs/mlos-3.13.yml
+++ b/conda-envs/mlos-3.13.yml
@@ -30,6 +30,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4
diff --git a/conda-envs/mlos-windows.yml b/conda-envs/mlos-windows.yml
index 906ba421c1..8e344d5592 100644
--- a/conda-envs/mlos-windows.yml
+++ b/conda-envs/mlos-windows.yml
@@ -31,6 +31,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4
diff --git a/conda-envs/mlos.yml b/conda-envs/mlos.yml
index 21335fa32e..4d8f82390c 100644
--- a/conda-envs/mlos.yml
+++ b/conda-envs/mlos.yml
@@ -26,6 +26,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4
diff --git a/mlos_bench/mlos_bench/launcher.py b/mlos_bench/mlos_bench/launcher.py
index 72701b83aa..0d76a25e6f 100644
--- a/mlos_bench/mlos_bench/launcher.py
+++ b/mlos_bench/mlos_bench/launcher.py
@@ -380,7 +380,10 @@ def add_argument(self, *args: Any, **kwargs: Any) -> None:
             default=False,
             dest="create_update_storage_schema_only",
             action="store_true",
-            help="Makes sure that the storage schema is update to date for the current version of mlos_bench.",
+            help=(
+                "Makes sure that the storage schema is up to date "
+                "for the current version of mlos_bench."
+            ),
         )
 
         # By default we use the command line arguments, but allow the caller to
diff --git a/mlos_bench/mlos_bench/storage/sql/alembic/versions/d2a708351ba8_add_alembic.py b/mlos_bench/mlos_bench/storage/sql/alembic/versions/d2a708351ba8_add_alembic.py
index 562a27dada..2974d80599 100644
--- a/mlos_bench/mlos_bench/storage/sql/alembic/versions/d2a708351ba8_add_alembic.py
+++ b/mlos_bench/mlos_bench/storage/sql/alembic/versions/d2a708351ba8_add_alembic.py
@@ -11,8 +11,8 @@
 """
 from collections.abc import Sequence
 
-import sqlalchemy as sa
-from alembic import op
+# import sqlalchemy as sa
+# from alembic import op
 
 # revision identifiers, used by Alembic.
 revision: str = "d2a708351ba8"