diff --git a/src/index.js b/src/index.js
index f467230b6..d5818b815 100644
--- a/src/index.js
+++ b/src/index.js
@@ -18,10 +18,9 @@ const noop = () => {};
 /** @typedef {import("webpack").Stats} Stats */
 /** @typedef {import("webpack").MultiStats} MultiStats */
 
-// TODO fix me
 /**
  * @typedef {Object} ExtendedServerResponse
- * @property {{ webpack?: { devMiddleware?: Context<any, any> } }} [locals]
+ * @property {{ webpack?: { devMiddleware?: Context<IncomingMessage, ServerResponse> } }} [locals]
  */
 
 /** @typedef {import("http").IncomingMessage} IncomingMessage */
diff --git a/types/index.d.ts b/types/index.d.ts
index 1ef3e3a77..ba4b67336 100644
--- a/types/index.d.ts
+++ b/types/index.d.ts
@@ -8,7 +8,7 @@ export = wdm;
 /** @typedef {import("webpack").MultiStats} MultiStats */
 /**
  * @typedef {Object} ExtendedServerResponse
- * @property {{ webpack?: { devMiddleware?: Context<any, any> } }} [locals]
+ * @property {{ webpack?: { devMiddleware?: Context<IncomingMessage, ServerResponse> } }} [locals]
  */
 /** @typedef {import("http").IncomingMessage} IncomingMessage */
 /** @typedef {import("http").ServerResponse & ExtendedServerResponse} ServerResponse */
@@ -186,7 +186,9 @@ type ExtendedServerResponse = {
     | {
         webpack?:
           | {
-              devMiddleware?: Context<any, any> | undefined;
+              devMiddleware?:
+                | Context<import("http").IncomingMessage, ServerResponse>
+                | undefined;
             }
           | undefined;
       }