fix bug

fix https://github.com/google/stagehand/issues/151