Fix build broken in previous build