Fixed issue in custom runner unit test