Fixed stevens if statement, thanks to steven