Fix missed code.