diff --git a/tool/btstack_memory_generator.py b/tool/btstack_memory_generator.py
index 7f4d0265f..59608726d 100755
--- a/tool/btstack_memory_generator.py
+++ b/tool/btstack_memory_generator.py
@@ -67,8 +67,8 @@ extern "C" {
 #include "classic/btstack_link_key_db_memory.h"
 #include "classic/rfcomm.h"
 #include "classic/sdp_server.h"
-#include "../test/avdtp/avdtp_sink.h"
-#include "../test/avdtp/avdtp_source.h"
+#include "classic/avdtp_sink.h"
+#include "classic/avdtp_source.h"
 
 // BLE
 #ifdef ENABLE_BLE