Fix brittle test