bugfix

minor changes