Fixed bug with setup script.

Hi there,
Just a tiny amend, as I was having a  problem with the setup script.  I hope that this is the right way of doing this, as I'm fairly new to forking & pull requests?
Feedback always welcome :smiley: 
Toby