Fixed small error in the script