Fix missing imports.