Fixes callback reference in updated code