[Fix] Fix minor issue introduced in previous commit