diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index c98b1507..d64e3b0b 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -47,7 +47,7 @@ jobs:
 
         OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
       run: |
-        poetry install -E dev -E postgres -E server
+        poetry install --all-extras
 
     - name: Set Poetry config
       env:
diff --git a/tests/utils.py b/tests/utils.py
index b80faa38..c08c0088 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -1,5 +1,4 @@
 import os
-import pexpect
 
 from memgpt.config import MemGPTConfig
 
@@ -18,6 +17,8 @@ def wipe_config():
 
 
 def configure_memgpt_localllm():
+    import pexpect
+
     wipe_config()
     child = pexpect.spawn("memgpt configure")