Apologies. I will send a new PR to fix this