Fixing the typo reported by voke in PR#8. Need to fix it on two lines.

He didn't fix the second line to make the tests pass, so I'm doing it here.