Fixes #5

This simple modification should resolve the issue.