Really amend code.
Merge pull request #3 from maddie/master

Fix derp.