Fix unit test count