ws tweaks

Whitespace tweaks - follow up to https://github.com/google/stagehand/pull/163.