Fix callback argument