diff --git a/Dockerfile b/Dockerfile
index a233002f..28c2c70a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -201,7 +201,7 @@ RUN set -ex; \
         cd /build/pgai; \
         for pg in ${PG_VERSIONS}; do \
             if [ "$pg" -gt 15 ]; then \
-                PG_MAJOR=${pg} make install; \
+                PG_MAJOR=${pg} ./projects/extension/build.py install; \
             fi; \
         done; \
     fi
diff --git a/Makefile b/Makefile
index 42d790a6..5f962c0a 100644
--- a/Makefile
+++ b/Makefile
@@ -11,7 +11,7 @@ PG_MAJOR?=16
 PG_VERSIONS?=
 
 # Additional PostgreSQL extensions we want to include with specific version/commit tags
-PGAI_VERSION?=extension-0.4.0
+PGAI_VERSION?=extension-0.4.1
 PGVECTORSCALE_VERSIONS?=all
 POSTGIS_VERSIONS?=3
 PG_AUTH_MON?=v2.0