diff --git a/applications/driver_unit_tests/command_stream.cpp b/applications/driver_unit_tests/command_stream.cpp
index a0b9171..ede9dd6 100644
--- a/applications/driver_unit_tests/command_stream.cpp
+++ b/applications/driver_unit_tests/command_stream.cpp
@@ -105,11 +105,11 @@
     ethosu_set_clock_and_power(&drv->dev, ETHOSU_CLOCK_Q_DISABLE, ETHOSU_POWER_Q_DISABLE);
 
     // Use simplified driver setup
-    ethosu_set_power_mode_v2(drv, true);
+    ethosu_set_power_mode(drv, true);
 }
 
 CommandStream::~CommandStream() {
-    ethosu_set_power_mode_v2(drv, false);
+    ethosu_set_power_mode(drv, false);
     ethosu_release_driver(drv);
 }
 
@@ -123,7 +123,7 @@
     }
 
     while (repeat-- > 0) {
-        int error = ethosu_invoke_v3(
+        int error = ethosu_invoke(
             drv, commandStream.data, commandStream.size, baseAddress, baseAddressSize, ETHOSU_DRIVER_BASEP_INDEXES);
 
         if (error != 0) {
diff --git a/targets/corstone-300/target.cpp b/targets/corstone-300/target.cpp
index 5d93854..2344995 100644
--- a/targets/corstone-300/target.cpp
+++ b/targets/corstone-300/target.cpp
@@ -135,7 +135,7 @@
 
 #ifdef ETHOSU
 void ethosuIrqHandler() {
-    ethosu_irq_handler();
+    ethosu_irq_handler(ethosu0_driver);
 }
 #endif
 
@@ -158,12 +158,12 @@
 
 #ifdef ETHOSU
     // Initialize Ethos-U NPU driver
-    if (ethosu_init_v4(ethosu0_driver,
-                       reinterpret_cast<void *>(ETHOSU_BASE_ADDRESS),
-                       ethosu_scratch,
-                       ETHOSU_FAST_MEMORY_SIZE,
-                       1,
-                       1)) {
+    if (ethosu_init(ethosu0_driver,
+                    reinterpret_cast<void *>(ETHOSU_BASE_ADDRESS),
+                    ethosu_scratch,
+                    ETHOSU_FAST_MEMORY_SIZE,
+                    1,
+                    1)) {
         printf("Failed to initialize NPU.\n");
         return;
     }