diff --git a/examples/dns/record.py b/examples/dns/record.py
index dc69faa87f9..72b89316a58 100644
--- a/examples/dns/record.py
+++ b/examples/dns/record.py
@@ -11,14 +11,15 @@
 
 record = client.dns.records.create(
     zone_id=zone_id,
-    type='A',
+    type="A",
     name="www.mydns.com",
     content="198.51.100.1",
-    proxied=True
+    proxied=True,
 )
+assert record is not None
 
 # clean up after we're done
 client.dns.records.delete(
     zone_id=zone_id,
-    dns_record_id=record['id']
+    dns_record_id=record.id,
 )