[IMP]minor fixing for new api.