Fixed bug with setup script.
Merge pull request #8 from CubicMushroom/setup-script-fix

Fixed bug with setup script.