diff --git a/include/cutlass/epilogue/fusion/sm90_visitor_compute_tma_warpspecialized.hpp b/include/cutlass/epilogue/fusion/sm90_visitor_compute_tma_warpspecialized.hpp
index 8f5ceb54..2ae10a68 100644
--- a/include/cutlass/epilogue/fusion/sm90_visitor_compute_tma_warpspecialized.hpp
+++ b/include/cutlass/epilogue/fusion/sm90_visitor_compute_tma_warpspecialized.hpp
@@ -78,7 +78,7 @@ using namespace detail;
 // the template argument.
 //
 // template<class A>
-// struct FooHomogeneous : public Foo<A, B> {};
+// struct FooHomogeneous : public Foo<A, A> {};
 //
 template<
   template <class> class ComputeFn,