Add missing imports to fix_absolute_import.py