#9 make tests more robust