diff --git a/document-ai/package.json b/document-ai/package.json
index 2e26f1e759..167966a7de 100644
--- a/document-ai/package.json
+++ b/document-ai/package.json
@@ -13,7 +13,7 @@
     "test": "mocha test/*.js --timeout 600000"
   },
   "dependencies": {
-    "@google-cloud/documentai": "^2.2.0",
+    "@google-cloud/documentai": "^2.2.1",
     "@google-cloud/storage": "^5.0.0",
     "p-queue": "^6.6.2",
     "uuid": "^8.3.1"