Minor fixes

2 minor tweaks