minor bugfix

some polishing