Fix the shell script