[REF][custom_fix] Fix relative import with 2to3 script