Fix missing imports