diff --git a/src/ethosu_device.c b/src/ethosu_device.c
index 273496d..7ce8fad 100644
--- a/src/ethosu_device.c
+++ b/src/ethosu_device.c
@@ -204,7 +204,7 @@
 #if !defined(ARM_NPU_STUB)
     uint32_t address = start_address;
 
-    ASSERT((start_address + num_reg) < NPU_IDS_REGISTERS_SIZE);
+    ASSERT((start_address + num_reg) < ID_REGISTERS_SIZE);
 
     for (int i = 0; i < num_reg; i++)
     {
@@ -422,7 +422,7 @@
 #if !defined(ARM_NPU_STUB)
     int i            = 0;
     uint32_t address = NPU_REG_SHARED_BUFFER0;
-    write_reg(NPU_REG_DEBUG_ADDR, section * BYTES_1KB);
+    write_reg(NPU_REG_DEBUG_ADDRESS, section * BYTES_1KB);
 
     while (address <= NPU_REG_SHARED_BUFFER255)
     {