Fix for failing test.