Could fix unit test failing