Fix bug

fixes #21