Fixed a missed update, following last commit