Fixed failing test for CI.