Update the code to invoke callback with the channel