bugfix
minor improvement