diff --git a/csrc/ck_tile_gemm_moe_2stages/gen_instances.py b/csrc/ck_tile_gemm_moe_2stages/gen_instances.py
index a80cb3f96d..c3a600a370 100644
--- a/csrc/ck_tile_gemm_moe_2stages/gen_instances.py
+++ b/csrc/ck_tile_gemm_moe_2stages/gen_instances.py
@@ -1,5 +1,5 @@
 # SPDX-License-Identifier: MIT
-# Copyright (C) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
+# Copyright (C) 2024-2026, Advanced Micro Devices, Inc. All rights reserved.
 import os
 import argparse
 from pathlib import Path
@@ -229,7 +229,7 @@ def fill_template(name, a_type, b_type, acc_type, c_type):
                 name=name, dtypes=f"{a_type}, {b_type}, {acc_type}, {c_type}"
             )
             if "fp4" in b_type:
-                body = "#if defined(__gfx950__)\n" + body + "\n#endif\n"
+                body = "#ifndef __gfx942__\n" + body + "\n#endif\n"
             Path(
                 os.path.join(
                     self.instances_path,