Fixed tests to use absolute import