Fix column reference at callback with a function