Fix unit tests to use relative imports