diff --git a/examples/cli/main.cpp b/examples/cli/main.cpp
index 8f68f53..d060404 100644
--- a/examples/cli/main.cpp
+++ b/examples/cli/main.cpp
@@ -653,7 +653,7 @@ void parse_args(int argc, const char** argv, SDParams& params) {
                 break;
             }
             params.ref_image_paths.push_back(argv[i]);
-        } else if (arg == "chroma-disable-dit-mask") {
+        } else if (arg == "--chroma-disable-dit-mask") {
             params.chroma_use_dit_mask = false;
         } else if (arg == "--chroma-use-t5-mask") {
             params.chroma_use_t5_mask = true;