diff --git a/document-ai/package.json b/document-ai/package.json
index ea52d7dd9c..f6b6088024 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": "^0.1.0",
+    "@google-cloud/documentai": "^1.0.0",
     "@google-cloud/storage": "^4.2.0"
   },
   "devDependencies": {