[Misc] Fix wrong code committed in previous commit