1 files changed, 4 insertions, 3 deletions
diff --git a/elf.py b/elf.py
index f0cc94e..0db1138 100755
--- a/elf.py
+++ b/elf.py
@@ -91,16 +91,17 @@ def elf(iter):
     segments = []
     segment = None
     for _, line in iter:
-        if line["type"] == "segment":
+        type = line["type"]
+        if type == "segment":
             (name, addr) = line["segment"]
             segment = {"name": name, "addr": addr, "content": []}
             segments.append(segment)
-        elif line["type"] == "instr":
+        elif type == "instr":
             if segment == None:
                 raise ValueError("label or code outside of segment!")
 
             segment["content"] += line["instr"]
-        else:
+        elif type != "empty":
             raise ValueError("elf input should contain only segments and data!")
 
     c = write_elf_header(segments)