diff --git a/index.d.ts b/index.d.ts
index 0b59dcb4..7a7ab378 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -90,7 +90,7 @@ declare module 'replicate' {
         webhook_events_filter?: WebhookEventType[];
         signal?: AbortSignal;
       },
-      progress?: (Prediction) => void
+      progress?: (prediction: Prediction) => void
     ): Promise<object>;
 
     request(route: string | URL, options: {
@@ -108,7 +108,7 @@ declare module 'replicate' {
         interval?: number;
         max_attempts?: number;
       },
-      stop?: (Prediction) => Promise<boolean>
+      stop?: (prediction: Prediction) => Promise<boolean>
     ): Promise<Prediction>;
 
     collections: {