diff --git a/document-ai/package.json b/document-ai/package.json
index 916bd06281..50fab5826f 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": "^1.0.0",
+    "@google-cloud/documentai": "^2.0.0",
     "@google-cloud/storage": "^5.0.0"
   },
   "devDependencies": {