diff --git a/index.d.ts b/index.d.ts
index a3e2ee0..d2402ca 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -110,7 +110,7 @@ declare module 'replicate' {
 
     wait(
       prediction: Prediction,
-      options: {
+      options?: {
         interval?: number;
       },
       stop?: (prediction: Prediction) => Promise<boolean>
diff --git a/package-lock.json b/package-lock.json
index bed1d6f..fb10a1e 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "replicate",
-  "version": "0.21.0",
+  "version": "0.21.1",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "replicate",
-      "version": "0.21.0",
+      "version": "0.21.1",
       "license": "Apache-2.0",
       "devDependencies": {
         "@types/jest": "^29.5.3",
diff --git a/package.json b/package.json
index 597df68..1b1d1ef 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "replicate",
-  "version": "0.21.0",
+  "version": "0.21.1",
   "description": "JavaScript client for Replicate",
   "repository": "github:replicate/replicate-javascript",
   "homepage": "https://github.com/replicate/replicate-javascript#readme",