PILOT-42 | fix import error from python3