just a quick tweak