Fix the callback error

#1