diff --git a/package.json b/package.json
index adc96b3..16396e6 100644
--- a/package.json
+++ b/package.json
@@ -3,6 +3,9 @@
   "version": "0.0.0",
   "description": "Koa HTTP adapter for Nest.js",
   "main": "dist/index.js",
+  "files": [
+    "dist"
+  ],
   "scripts": {
     "test": "mocha",
     "build:publish": "tsc --project tsconfig.publish.json",
diff --git a/src/NestKoaMiddleware.ts b/src/NestKoaMiddleware.ts
index 5a0ddbe..ca6ecd2 100644
--- a/src/NestKoaMiddleware.ts
+++ b/src/NestKoaMiddleware.ts
@@ -11,7 +11,7 @@ export interface NestKoaMiddleware {
 }
 
 export const koaToNestMiddleware = (
-  middleware: Koa.Middleware,
+  middleware: Koa.Middleware<any, any>,
 ): NestKoaFunctionalMiddleware => (
   req: Koa.Request,
   res: Koa.Response,