[fix] use proper callback