Fix python 3 test import