Add back in the callbacks that were removed