Index: tests/test_dg_system_record.py
===================================================================
diff -u -r97c98d84604e3700dd446235e91a5ebe2db08022 -r7cd740bfb30cafa48847843cd09c3ee346b882a4
--- tests/test_dg_system_record.py	(.../test_dg_system_record.py)	(revision 97c98d84604e3700dd446235e91a5ebe2db08022)
+++ tests/test_dg_system_record.py	(.../test_dg_system_record.py)	(revision 7cd740bfb30cafa48847843cd09c3ee346b882a4)
@@ -12,15 +12,15 @@
     sleep(2)
 
     sys = dg.system_record
-    #sys.dg_system_record['system_record']['top_level_pn'][1] = 'DEN-SN-Y20'
+    sys.dg_system_record['system_record']['top_level_pn'][1] = 'DEN-SN-Y20'
     print(sys.dg_system_record)
     test = True
 
     if test:
         sys.get_dg_system_record()
         while True:
             sleep(0.2)
-            if sys.current_message == sys.total_messages:
+            if sys.is_reading_record_done():
                 break
     else:
         sys.set_dg_system_record()