Fix callback function reference