diff --git a/mgz/fast/header.py b/mgz/fast/header.py index f96c878..c583cef 100644 --- a/mgz/fast/header.py +++ b/mgz/fast/header.py @@ -230,10 +230,10 @@ def parse_scenario(data, num_players, version, save): """Parse scenario section.""" data.read(4455) if version is Version.DE: - data.read(128) + data.read(102) + scenario_filename = aoc_string(data) + data.read(24) instructions = aoc_string(data) - for _ in range(0, 9): - aoc_string(data) data.read(78) for _ in range(0, 16): aoc_string(data) @@ -278,7 +278,8 @@ def parse_scenario(data, num_players, version, save): return dict( map_id=map_id, difficulty_id=difficulty_id, - instructions=instructions + instructions=instructions, + scenario_filename=scenario_filename )