Fix reply callback bug