Bug fix

fixes #54