small script cleanup