Fixed small script issue