Deflaking tests

Signed-off-by: Andreas Tolfsen <ato@mozilla.com>