fix the script a little to at least work :)