fixing error in #2

This is a pull request for issue #2