bug fix

Reviewed By: VLanvin Differential Revision: D57249553 fbshipit-source-id: 813889db5d591a9b72688648d2e0cdd26880fa67
WhatsApp · May 13, 2024 · 266a1dd · 266a1dd
1 parent ec3c430
commit 266a1dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eqwalizer/src/main/scala/com/whatsapp/eqwalizer/tc/Check.scala b/eqwalizer/src/main/scala/com/whatsapp/eqwalizer/tc/Check.scala
@@ -542,7 +542,7 @@ final class Check(pipelineContext: PipelineContext) {
     val (argTys, env1) = elab.elabExprs(args, env)
     var ftResTy = elabApply.elabApply(ft, args, argTys, env1)
     if (CustomReturn.isCustomReturn(funId))
-      ftResTy = CustomReturn.customizeResultType(funId, args, resTy)
+      ftResTy = CustomReturn.customizeResultType(funId, args, ftResTy)
     if (!subtype.subType(ftResTy, resTy))
       diagnosticsInfo.add(ExpectedSubtype(expr.pos, expr, expected = resTy, got = ftResTy))
     env1