Fix unit test extension