Adjust test to fix previous change