another quick tweak for cleanup