diff --git a/docker-compose.yml b/docker-compose.yml
index 6eabf2aa..b68bbbea 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,31 +1,9 @@
-# Use to test langchain-postgres
-version: "3"
 name: langchain-postgres
 
 services:
-  postgres:
-    image: postgres:16
-    environment:
-      POSTGRES_DB: langchain
-      POSTGRES_USER: langchain
-      POSTGRES_PASSWORD: langchain
-    ports:
-      - "6023:5432"
-    command: |
-      postgres -c log_statement=all
-    healthcheck:
-      test:
-        [
-          "CMD-SHELL",
-          "psql postgresql://langchain:langchain@localhost/langchain --command 'SELECT 1;' || exit 1",
-        ]
-      interval: 5s
-      retries: 60
-    volumes:
-      - postgres_data:/var/lib/postgresql/data
   pgvector:
     # postgres with the pgvector extension
-    image: ankane/pgvector
+    image: pgvector/pgvector:pg16
     environment:
       POSTGRES_DB: langchain
       POSTGRES_USER: langchain
@@ -43,8 +21,7 @@ services:
       interval: 5s
       retries: 60
     volumes:
-      - postgres_data_pgvector:/var/lib/postgresql/data
+      - postgres_data_pgvector_16:/var/lib/postgresql/data
 
 volumes:
-  postgres_data:
-  postgres_data_pgvector:
+  postgres_data_pgvector_16: