diff --git a/src/glsl/link_functions.cpp b/src/glsl/link_functions.cpp
index a9ed49a3492..327be73afe6 100644
--- a/src/glsl/link_functions.cpp
+++ b/src/glsl/link_functions.cpp
@@ -164,6 +164,8 @@ public:
        */
       linked_sig->accept(this);
 
+      ir->set_callee(linked_sig);
+
       return visit_continue;
    }