Clean up an import in domains.py