diff --git a/example/01_gemm/common.hpp b/example/01_gemm/common.hpp
index 9073ffcfc1..4104e83677 100644
--- a/example/01_gemm/common.hpp
+++ b/example/01_gemm/common.hpp
@@ -57,6 +57,7 @@ struct ProblemSizeStreamK_universal final
     ck::index_t StrideB = -1;
     ck::index_t StrideC = -1;
 
+    ck::index_t NumSKBlocks = -1; // number of stream-k blocks
     ck::index_t Grid_size   = -1; // defaults to max occupancy
     ck::index_t Streamk_sel = 1;  // defaults to 1-tile SK
 };
diff --git a/example/01_gemm/run_gemm_example_streamk_v2.inc b/example/01_gemm/run_gemm_example_streamk_v2.inc
index 439dd4504f..f9f2f2cb89 100644
--- a/example/01_gemm/run_gemm_example_streamk_v2.inc
+++ b/example/01_gemm/run_gemm_example_streamk_v2.inc
@@ -150,8 +150,7 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
         StrideC,
         a_element_op,
         b_element_op,
-        c_element_op
-        /**problem_size.NumSKBlocks**/); // TODO: add NumSKBlocks arg into Problem?
+        c_element_op problem_size.NumSKBlocks); // TODO: add NumSKBlocks arg into Problem?
 
     if(!gemm.IsSupportedArgument(argument))
     {