Fix to failing test.