Bug fix for failing test.