Correct the bug for main caller