Fix missed files caused by previous commit