Fixed runtime error

My bad. My code broke your script. This should fix it.
Merge pull request #9 from 0gb-us/patch-2

Fixed runtime error