Fix one more test