Fix one more failing test