diff --git a/scripts/assemble.py b/scripts/assemble.py
index cd0b966..265a78e 100755
--- a/scripts/assemble.py
+++ b/scripts/assemble.py
@@ -50,7 +50,7 @@
     def find_slots(self, bootdir):
         offsets = {}
         sizes = {}
-        with open(os.path.join(bootdir, 'zephyr', 'include', 'generated', 'generated_dts_board_unfixed.h'), 'r') as fd:
+        with open(os.path.join(bootdir, 'zephyr', 'include', 'generated', 'devicetree_unfixed.h'), 'r') as fd:
             for line in fd:
                 m = offset_re.match(line)
                 if m is not None:
@@ -60,7 +60,7 @@
                     sizes[m.group(1)] = int(m.group(3), 0)
 
         if not same_keys(offsets, sizes):
-            raise Exception("Inconsistent data in generated_dts_board.h")
+            raise Exception("Inconsistent data in devicetree.h")
 
         # We care about the MCUBOOT, IMAGE_0, and IMAGE_1 partitions.
         if 'MCUBOOT' not in offsets: