bugfix, small improvement