Fix test breaks.
2 files changed