Bug fix

Fixes bug #15.