fixes #3

This is a hot fix for issue #3
Awesome job! Go ahead and merge this back into master.