diff --git a/baddns/modules/references.py b/baddns/modules/references.py index 30d60ed..f731acd 100644 --- a/baddns/modules/references.py +++ b/baddns/modules/references.py @@ -166,7 +166,7 @@ async def process_cname_analysis(self, parsed_results): "direct_mode": direct_mode, } cname_findings.append(finding) - cname_instance.cleanup() + await cname_instance.cleanup() return cname_findings async def dispatch(self): diff --git a/baddns/modules/txt.py b/baddns/modules/txt.py index 2024f30..daa3adb 100644 --- a/baddns/modules/txt.py +++ b/baddns/modules/txt.py @@ -60,7 +60,7 @@ async def dispatch(self): "trigger": self.target_dnsmanager.target, } ) - cname_instance_direct.cleanup() + await cname_instance_direct.cleanup() cname_instance = BadDNS_cname( host, @@ -79,7 +79,7 @@ async def dispatch(self): "trigger": self.target_dnsmanager.target, } ) - cname_instance.cleanup() + await cname_instance.cleanup() return True