diff --git a/cloudfail.py b/cloudfail.py index ba0edde..eecc704 100644 --- a/cloudfail.py +++ b/cloudfail.py @@ -177,9 +177,9 @@ def check_for_wildcard(target): answer = resolver.query('*.' + target) #If found, ask user if continue as long until valid answer choice = '' - while choice is not 'y' and choice is not 'n': + while choice != 'y' and choice != 'n': choice = input("A wildcard DNS entry was found. This will result in all subdomains returning an IP. Do you want to scan subdomains anyway? (y/n): ") - if choice is 'y': + if choice == 'y': return False else: return True