test.py - change the way imports are made.