1 files changed, 3 insertions, 3 deletions
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 @@ CommandStream::CommandStream(const DataPointer &_commandStream,
     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 @@ int CommandStream::run(size_t repeat) {
     }
 
     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) {