Tweaks + Fixes

Change-Id: Ie9bb66222a1912489697939034d735b4d4998156