Fix missed include