Fix issue #3

Thanks to @wprater for submitting this issue.