[Fix] - Correct the type