[Fix] Fix issue introduced by previous commit