minor fix to optimize test case