small improvement

small improvement