Add a return after running a callback