make it work with koa