Author here. Thanks for the correction.

You caught me red-handed for not testing these codes in playground.