Fix problem with shell script