diff --git a/cli.js b/cli.js
index b90d05e..3b7744d 100755
--- a/cli.js
+++ b/cli.js
@@ -81,8 +81,7 @@ const cli = yargs(process.argv.slice(2))
     },
   });
 
-// TODO: checks for any .css arg, instead of parsing --css
-const css = process.argv.find((arg) => /\.css$/.test(arg)) || "gmi-web.css";
+const css = yargs(process.argv.slice(2)).exitProcess(false).argv.css || "gmi-web.css";
 const CSS_VARS = CSS.rootVariables(CSS.load({ css }));
 Object.keys(CSS_VARS).forEach((key) => {
   const opt = key.replace("--", "");