diff --git a/enzyme/enzyme-tests.tsx b/enzyme/enzyme-tests.tsx
index e5d56200cb..b87a759861 100644
--- a/enzyme/enzyme-tests.tsx
+++ b/enzyme/enzyme-tests.tsx
@@ -204,7 +204,7 @@ namespace ShallowWrapperTest {
     }
 
     function test_state() {
-        shallowWrapper.state();
+        const state: MyComponentState = shallowWrapper.state();
         shallowWrapper.state('key');
         const tmp: String = shallowWrapper.state<String>('key');
     }
diff --git a/enzyme/enzyme.d.ts b/enzyme/enzyme.d.ts
index 5a53f8bf20..85a72127c1 100644
--- a/enzyme/enzyme.d.ts
+++ b/enzyme/enzyme.d.ts
@@ -204,8 +204,9 @@ declare module "enzyme" {
          * Returns the state hash for the root node of the wrapper. Optionally pass in a prop name and it will return just that value.
          * @param [key]
          */
-        state(key?: string): any;
-        state<T>(key?: string): T;
+        state(): S;
+        state(key: string): any;
+        state<T>(key: string): T;
 
         /**
          * Returns the context hash for the root node of the wrapper. Optionally pass in a prop name and it will return just that value.