Find and replace poly types for call expr

1 files changed, 6 insertions, 0 deletions

diff --git a/src/server/generics.odin b/src/server/generics.odin
index 6f0be73..e8abdaf 100644
--- a/src/server/generics.odin
+++ b/src/server/generics.odin
@@ -463,6 +463,12 @@ find_and_replace_poly_type :: proc(expr: ^ast.Expr, poly_map: ^map[string]^ast.E
 				v.pos.file = expr.pos.file
 				v.end.file = expr.end.file
 			}
+		case ^ast.Call_Expr:
+			for &arg in v.args {
+				if expr, ok := get_poly_map(arg, poly_map); ok {
+					arg = expr
+				}
+			}
 		}
 
 		return visitor