Fix importerror catching in pypy