i think you can add that small change to this pr