Added missing imports to headcanon