Fix relative imports in tests