[Fix] - Correct the bug