Problem with import with PT2