small improvement ^^