I've tested your patch and it looks fine. Do you want to send a PR ?
Otherwise I can just push your patch myself.