diff --git a/UndertaleModLib/Decompiler/Decompiler.cs b/UndertaleModLib/Decompiler/Decompiler.cs
index 58f6cb6e2..d5aa564b7 100644
--- a/UndertaleModLib/Decompiler/Decompiler.cs
+++ b/UndertaleModLib/Decompiler/Decompiler.cs
@@ -217,7 +217,7 @@ internal static void DecompileFromBlock(DecompileContext context, Dictionary<uin
                         if (!(nextBlock is not null
                             && nextBlock.Instructions.Count > 0
                             && nextBlock.Instructions[0].Kind == UndertaleInstruction.Opcode.Push
-                            && nextBlock.Instructions[0].Value.GetType() != typeof(int)))
+                            && nextBlock.Instructions[0].Value is UndertaleInstruction.Reference<UndertaleFunction>))
                         {
                             ReturnStatement stmt = new ReturnStatement(instr.Kind == UndertaleInstruction.Opcode.Ret ? stack.Pop() : null);
                             /*