diff --git a/.github/workflows/pages.yml b/.github/workflows/pages.yml
index d0306df..cb09374 100644
--- a/.github/workflows/pages.yml
+++ b/.github/workflows/pages.yml
@@ -48,8 +48,8 @@ jobs:
           npm install @redocly/cli -g
       - name: Export OpenAPI docs
         run: |
-          python main.py export-openapi --export-path "openapi-kobold.json"
-          python main.py export-openapi --export-path "openapi-oai.json"
+          python main.py export-openapi --export-path "openapi-kobold.json" --api-servers kobold
+          python main.py export-openapi --export-path "openapi-oai.json" --api-servers oai
       - name: Build and store Redocly site
         run: |
           mkdir static
diff --git a/start.py b/start.py
index 8e95829..8fa46e6 100644
--- a/start.py
+++ b/start.py
@@ -272,7 +272,6 @@ def migrate_gpu_lib():
 
     # Assume all dependencies are installed from here
     try:
-        from common.args import convert_args_to_dict
         from main import entrypoint
 
         # Create a config if it doesn't exist