bugfix

working on #6