Fix build, address missed comment