diff --git a/src/nodes/node_conf.rs b/src/nodes/node_conf.rs
index 4fb8f50..422e8be 100644
--- a/src/nodes/node_conf.rs
+++ b/src/nodes/node_conf.rs
@@ -703,6 +703,11 @@ impl NodeConfigurator {
     /// updates are then sent to the audio thread.
     /// See also [NodeConfigurator::get_block_function].
     pub fn check_block_function(&mut self, id: usize) -> Result<(), BlkJITCompileError> {
+        #[cfg(feature = "synfx-dsp-jit")]
+        if let Some(cod) = self.code_engines.get_mut(id) {
+            cod.query_returns();
+        }
+
         #[cfg(feature = "synfx-dsp-jit")]
         if let Some((generation, block_fun)) = self.block_functions.get_mut(id) {
             if let Ok(block_fun) = block_fun.lock() {