couple tweaks I forgot