diff --git a/docker-compose.yml b/docker-compose.yml
index 2fda5490..c2f2e250 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -13,7 +13,7 @@ services:
       - db_network
 
   xata-agent:
-    image: xataio/agent:0.3.1
+    image: xataio/agent:0.4.0
     restart: unless-stopped
     environment:
       DATABASE_URL: postgresql://${POSTGRES_USER:-dbagent}:${POSTGRES_PASSWORD:-changeme}@postgres:5432/${POSTGRES_DB:-dbagent}