diff --git a/python/pystencils_walberla/kernel_selection.py b/python/pystencils_walberla/kernel_selection.py
index 0e2becd6f5efdbced366cdaed5b2d4ec536ae3e1..316c701c1ec837c399a8a39f2d802ad1151fb42d 100644
--- a/python/pystencils_walberla/kernel_selection.py
+++ b/python/pystencils_walberla/kernel_selection.py
@@ -108,8 +108,8 @@ class AbstractConditionNode(AbstractKernelSelectionNode):
         true_branch_code = self.branch_true.get_code(**kwargs)
         false_branch_code = self.branch_false.get_code(**kwargs)
 
-        true_branch_code = do_indent(true_branch_code, width=4, indentfirst=True)
-        false_branch_code = do_indent(false_branch_code, width=4, indentfirst=True)
+        true_branch_code = do_indent(true_branch_code, width=4, first=True)
+        false_branch_code = do_indent(false_branch_code, width=4, first=True)
 
         code = f"if({self.condition_text}) {{\n"
         code += true_branch_code