Fix flaky test.
Merge pull request #65 from PencilCode/fixtest

Fix flaky test.