Couple tweaks