Ah yeah. I forgot to review the comments before making the PR :) Will update ...