Fixed #11

This pull request is a suggested fix for #11.