Minor tweak to shell script