fixes

[m6w6/ext-psi] / src / validator.c

diff --git a/src/validator.c b/src/validator.c
index 7bc2f701989c698c63a0e491f4031ab38b30dd19..510bba1ad7280ceb0b262177f77f57c1b289ab5f 100644 (file)
--- a/src/validator.c
+++ b/src/validator.c
@@ -266,11 +266,13 @@ static inline int validate_impl_func(PSI_Validator *V, impl *impl, impl_func *fu
        }
        return 1;
 }
+
 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
        size_t i;
 
        for (i = 0; i < decls->count; ++i) {
                if (!strcmp(decls->list[i]->func->var->name, ret->decl->name)) {
+                       ret->decl->arg = decls->list[i]->func;
                        return decls->list[i];
                }
        }