[WL-135] Fixed a flaky test. Added a missing assertion in a test.