Use python version dependent import

Change-Id: Iae6bc0cc8d526162b91d0c18cf1fba1461aa9f98