small improvement on previous commit