Fix of unit test