Fix bug

Fixes issue #1