minor tweaks
forgot this