X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fvalidator.c;fp=src%2Fvalidator.c;h=510bba1ad7280ceb0b262177f77f57c1b289ab5f;hp=7bc2f701989c698c63a0e491f4031ab38b30dd19;hb=a7a2bb568d3ac79ce398941ab9231f0f9555bb22;hpb=d60f4bbdd315ddf42dbafeff0fd3d87d2e7a51f7

diff --git a/src/validator.c b/src/validator.c
index 7bc2f70..510bba1 100644
--- 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];
 		}
 	}