[FIX] small improvement