fix shell script a bit