Fixed TMATS instantiation.

atac · Jun 15, 2021 · 92ed7d2 · 92ed7d2
1 parent b6580ef
commit 92ed7d2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chapter10/computer.py b/chapter10/computer.py
@@ -58,8 +58,8 @@ def __init__(self, *args, **kwargs):
         packet.Packet.__init__(self, *args, **kwargs)
         if self.buffer:
             self.data = self.buffer.read(self.data_length - 4)
-        elif not self.data:
-            self.data = bytes()
+        elif not getattr(self, 'data', None):
+            self.data = kwargs.get('data', bytes())
 
     def _raw_body(self):
         csdw = self.csdw_format.pack(self.__dict__)