Fix unit test discovery