Fixes #5

This should fix issue #5.