Update script.py

Just little fix