diff --git a/types/mobx-task/index.d.ts b/types/mobx-task/index.d.ts
index 0a160a4b70..b5b1a87d81 100644
--- a/types/mobx-task/index.d.ts
+++ b/types/mobx-task/index.d.ts
@@ -37,13 +37,14 @@ export interface TaskStatusAware<Result = any, Args extends any[] = any[]> exten
 
     wrap<R, A extends any[]>(func: (inner: TaskStatusAware<Result, Args>) => (...args: A) => Promise<R>): TaskStatusAware<R, A>;
 
-    setState(props: { state: TaskState, result: Result }): void;
+    setState(props: TaskOptions<Result>): void;
 
     reset(): void;
 }
 
 export interface TaskCreator<K extends keyof TaskOptions<any>> extends MethodDecorator, PropertyDecorator {
     <R, A extends any[]>(func: TaskFunc<R, A>, options?: Pick<TaskOptions<WithoutPromise<R>>, K>): TaskStatusAware<WithoutPromise<R>, A>;
+
     (options: Pick<TaskOptions<WithoutPromise<any>>, K>): PropertyDecorator | MethodDecorator;
 }