Fix the callback