Fix for final broken test.