1 files changed, 4 insertions, 0 deletions
diff --git a/src/checker.cpp b/src/checker.cpp
index 8f0cc1cd1..852fb89bb 100644
--- a/src/checker.cpp
+++ b/src/checker.cpp
@@ -1479,6 +1479,10 @@ gb_internal Entity *entity_of_node(Ast *expr) {
 	case_ast_node(cc, CaseClause, expr);
 		return cc->implicit_entity;
 	case_end;
+
+	case_ast_node(ce, CallExpr, expr);
+		return ce->entity_procedure_of;
+	case_end;
 	}
 	return nullptr;
 }