small tweak to ruby script