[fix minor] Correct test race condition introduced by #226.