diff --git a/teapot/passes/transient/transient_indirect_branch_check_dest_pass.py b/teapot/passes/transient/transient_indirect_branch_check_dest_pass.py
index a46fb84..15ddac4 100644
--- a/teapot/passes/transient/transient_indirect_branch_check_dest_pass.py
+++ b/teapot/passes/transient/transient_indirect_branch_check_dest_pass.py
@@ -45,7 +45,7 @@ def visit_code_block(self, block: gtirb.CodeBlock, function: Function = None):
             instructions = list(self.decoder.get_instructions(block))
             last_inst = instructions[-1]
 
-            if last_inst.mnemonic == "ret":
+            if non_fallthrough_edges[0].label.type == gtirb.cfg.Edge.Type.Return:
                 operand_str = "[rsp]"
             else:
                 dest_operand = last_inst.operands[0]