FIxed #266 -- attempt to deflake our tests (#660)

* FIxed #266 -- attempt to deflake our tests

* typo
2 files changed