fix to test function.