Fix one of the tests