quick tweak to clean script