diff --git a/example/ck_tile/21_elementwise/elementwise_common.hpp b/example/ck_tile/21_elementwise/elementwise_common.hpp
index fb6af4d3e0..fd7889bd35 100644
--- a/example/ck_tile/21_elementwise/elementwise_common.hpp
+++ b/example/ck_tile/21_elementwise/elementwise_common.hpp
@@ -1,6 +1,7 @@
 // SPDX-License-Identifier: MIT
 // Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
 
+#include <variant>
 #include "ck_tile/core/arch/arch.hpp"
 
 auto string_to_datatype(const std::string& datatype)
diff --git a/example/ck_tile/21_elementwise/elementwise_example_unary.cpp b/example/ck_tile/21_elementwise/elementwise_example_unary.cpp
index a1804ea52e..54533e01b2 100644
--- a/example/ck_tile/21_elementwise/elementwise_example_unary.cpp
+++ b/example/ck_tile/21_elementwise/elementwise_example_unary.cpp
@@ -158,7 +158,14 @@ bool filter_then_run(const ck_tile::ArgParser& arg_parser)
     bool pass = true;
 
     if constexpr(std::is_same_v<XElementwiseOperation, ck_tile::element_wise::UnarySquare> &&
-                 std::is_same_v<XDataType, ck_tile::bf16_t>)
+                 (std::is_same_v<XDataType, ck_tile::bf16_t> ||
+                  std::is_same_v<YDataType, ck_tile::bf16_t>))
+    {
+        throw_unsupported();
+    }
+    else if constexpr(std::is_same_v<XElementwiseOperation, ck_tile::element_wise::UnaryConvert> &&
+                      (std::is_same_v<XDataType, ck_tile::bf16_t> ||
+                       std::is_same_v<YDataType, ck_tile::bf16_t>))
     {
         throw_unsupported();
     }