[fix test] Make the tests pass again.