[Fix] - Add the missing parameters