a couple of small tweaks