Reinstated correction that I accidentally overwrote.
Merge pull request #2 from travisbrown/master

Correction correction