minor fixes

1 files changed, 3 insertions, 2 deletions

diff --git a/ascension/ascension.py b/ascension/ascension.py
index 4d521b4..d1492f8 100644
--- a/ascension/ascension.py
+++ b/ascension/ascension.py
@@ -269,6 +269,8 @@ class Ascender():
         # Check if there is zone has already been migrated
         nsrecords = cls.zone.iterate_rdatas(dns.rdatatype.NS)
 
+        # This is broken if your NS is for ns.foo.YOURZONE as you add
+        # the PKEY to YOURZONE instead of to the foo.YOURZONE subzone.
         gnspkey = list(filter(lambda record: str(record[2]).startswith('gns--pkey--'), nsrecords))
         if gnspkey:
             label = str(gnspkey[0][0])
@@ -342,8 +344,7 @@ class Ascender():
             logging.critical("thread join timed out, still running")
 
         # Add soa record to GNS once completed (updates the previous one)
-        soa = cls.get_zone_soa(cls.zone)
-        cls.add_soa_record_to_gns(soa)
+        cls.add_soa_record_to_gns(cls.soa)
         logging.info("All records have been added!")
 
     @staticmethod