1 files changed, 6 insertions, 2 deletions
diff --git a/src/cli.ts b/src/cli.ts
index 0a5c9612d..9d1815256 100644
--- a/src/cli.ts
+++ b/src/cli.ts
@@ -5,10 +5,14 @@ import { promises as fsPromises } from 'fs';
 import yargs from 'yargs-parser';
 
 import { loadConfig } from './config.js';
-import generate from './generate.js';
+import {build} from './build.js';
 import devServer from './dev.js';
 
 const { readFile } = fsPromises;
+const buildAndExit = async (...args: Parameters<typeof build>) => {
+  const ret = await build(...args);
+  process.exit(ret);
+}
 
 type Arguments = yargs.Arguments;
 type cliState = 'help' | 'version' | 'dev' | 'build';
@@ -61,7 +65,7 @@ async function runCommand(rawRoot: string, cmd: (a: AstroConfig) => Promise<void
 }
 
 const cmdMap = new Map([
-  ['build', generate],
+  ['build', buildAndExit],
   ['dev', devServer],
 ]);