Yep, completely missed that on the refactor. Will update.