Bug fix

This fixes #5.