Fixed a callback bug