Fix the test as a result of 25cc979