diff --git a/python/cutlass/op/gemm.py b/python/cutlass/op/gemm.py
index c78033bd..e486a431 100644
--- a/python/cutlass/op/gemm.py
+++ b/python/cutlass/op/gemm.py
@@ -403,7 +403,7 @@ class Gemm(OperationBase):
         """
         tds = [datatypes.td_from_profiler_op(op) for op in self.possible_operations.all_operations]
         if self._math_operation is not None:
-            tds = [td for td in tds if td.tile_description.math_instruction == self._math_operation]
+            tds = [td for td in tds if td.math_instruction.math_operation == self._math_operation]
         return tds
 
     def construct(
@@ -712,4 +712,4 @@ class Gemm(OperationBase):
         if sync:
             arguments.sync()
 
-        return arguments
+        return arguments
\ No newline at end of file