diff --git a/tools/unit-tests/unit-update-disk-fs.c b/tools/unit-tests/unit-update-disk-fs.c
index 287b23eb..98f06ac8 100644
--- a/tools/unit-tests/unit-update-disk-fs.c
+++ b/tools/unit-tests/unit-update-disk-fs.c
@@ -682,9 +682,10 @@ uint32_t wolfBoot_get_blob_version(uint8_t *blob)
     return version;
 }
 
-int wolfBoot_get_dts_size(void *dts_addr)
+int wolfBoot_get_dts_size(void *dts_addr, uint32_t capacity)
 {
     (void)dts_addr;
+    (void)capacity;
     return -1;
 }