[IMP] small improvement