Fix second unit test