Added back relative import to the tests.