Fix lost include