couple of small tweaks