small improvement (one line)