Fixes #1

Check out the change I made. You decide if my pull request is approved or not.