diff --git a/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.kt b/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.kt
index 5d4ba1c4..beb710a7 100644
--- a/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.kt
+++ b/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.kt
@@ -214,7 +214,7 @@ class IfElseTransformer : Transformer() {
         return getIf() != null
     }
 
-    private fun Statement.getIf(): Statement? {
+    private fun Statement.getIf(): IfStmt? {
         return when (this) {
             is IfStmt -> clone()
             is BlockStmt -> {