diff --git a/transformer_engine/jax/cpp_extensions/gemm.py b/transformer_engine/jax/cpp_extensions/gemm.py
index 68c72d5059..3e2df50833 100644
--- a/transformer_engine/jax/cpp_extensions/gemm.py
+++ b/transformer_engine/jax/cpp_extensions/gemm.py
@@ -895,8 +895,8 @@ def _parse_operand_output_specs(
         if gsr.tp_resource is not None:
             if gsr.tp_resource in lhs_specs:
                 warnings.warn(
-                    "Tensor sequence parallelism is detected as tp_resource='{gsr.tp_resource}'"
-                    " appears in lhs_specs: {lhs_specs}. Please setting MeshResource.tpsp_resource"
+                    f"Tensor sequence parallelism is detected as tp_resource='{gsr.tp_resource}'"
+                    f" appears in lhs_specs: {lhs_specs}. Please setting MeshResource.tpsp_resource"
                     " for tensor sequence parallelism to avoid potential issues."
                 )