Fix one unit test