diff --git a/packages/typegen/src/fromChain.ts b/packages/typegen/src/fromChain.ts
index 9c20349fb1..1885ecd0c3 100644
--- a/packages/typegen/src/fromChain.ts
+++ b/packages/typegen/src/fromChain.ts
@@ -27,7 +27,7 @@ function generate (metaHex: string, pkg: string | undefined, output: string, isS
 
   writeFile(path.join(process.cwd(), output, 'augment-api.ts'), (): string =>
     [
-      HEADER,
+      HEADER('chain'),
       ...[
         '@polkadot/api/augment/rpc',
         ...['consts', 'query', 'tx'].filter((key) => !!key).map((key) => `./augment-api-${key}`)