diff --git a/cli.ts b/cli.ts
index d9d37340..5c75723e 100644
--- a/cli.ts
+++ b/cli.ts
@@ -14,7 +14,7 @@ const upgrade = new Command()
   .example("lume upgrade -g", "Upgrades to the latest stable version.")
   .example("lume upgrade --dev", "Upgrades to the latest development version.")
   .action(async ({ dev, version }) => {
-    const { upgrade } = await import("./cli/upgrade.ts");
+    const { default: upgrade } = await import("./cli/upgrade.ts");
     await upgrade(dev, version);
   });
 
diff --git a/cli/upgrade.ts b/cli/upgrade.ts
index 175bcf91..2e5ec756 100644
--- a/cli/upgrade.ts
+++ b/cli/upgrade.ts
@@ -1,11 +1,6 @@
 import { upgrade } from "../deps/init.ts";
 
-interface Options {
-  dev?: boolean;
-  version?: string;
-}
-
-export default function ({ dev, version }: Options) {
+export default async function (dev?: boolean, version?: string) {
   const process = upgrade({ path: ".", dev, version });
-  process.run();
+  await process.run();
 }