diff --git a/atroposlib/envs/server_handling/server_manager.py b/atroposlib/envs/server_handling/server_manager.py
index 433c0f19..36d6440b 100644
--- a/atroposlib/envs/server_handling/server_manager.py
+++ b/atroposlib/envs/server_handling/server_manager.py
@@ -90,9 +90,9 @@ class ServerManager:
                         api_key="x",
                     )
                 )
-            self.servers = [OpenAIServer(config) for config in openai_configs]
+            self.servers = [server_class(config) for config in openai_configs]
         elif not slurm:
-            self.servers = [OpenAIServer(config) for config in configs]
+            self.servers = [server_class(config) for config in configs]
         else:
             nodelist = (
                 os.popen(f'scontrol show hostnames {os.environ["SLURM_JOB_NODELIST"]}')
diff --git a/atroposlib/envs/server_handling/trl_vllm_server.py b/atroposlib/envs/server_handling/trl_vllm_server.py
index 2dc18dd6..2007fc9c 100644
--- a/atroposlib/envs/server_handling/trl_vllm_server.py
+++ b/atroposlib/envs/server_handling/trl_vllm_server.py
@@ -1,5 +1,5 @@
 """
-This is a server that interfaces with trl's vLLM server."
+This is a server that interfaces with trl's vLLM server.
 
 Developed with much help from @winglian when they worked on integrating Atropos into Axolotl.
 """
diff --git a/environments/gsm8k_server axolotl.py b/environments/gsm8k_server_axolotl.py
similarity index 100%
rename from environments/gsm8k_server axolotl.py
rename to environments/gsm8k_server_axolotl.py