Fix build broken by last commit