Fix fragile test