diff --git a/python/triton/code_gen.py b/python/triton/code_gen.py
index e4b7d9f0f..deede2530 100644
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -279,7 +279,7 @@ class CodeGenerator(ast.NodeVisitor):
 
     def visit_IfExp(self, node):
         cond = self.visit(node.test)
-        if cond:
+        if cond.value:
             return self.visit(node.body)
         else:
             return self.visit(node.orelse)