FIX forgot to import pycompat