diff --git a/test/DriverTestHelpers.cpp b/test/DriverTestHelpers.cpp
index c6d4406..8e8d7be 100644
--- a/test/DriverTestHelpers.cpp
+++ b/test/DriverTestHelpers.cpp
@@ -198,7 +198,8 @@
 
 android::sp<V1_3::IPreparedModel> PrepareModelWithStatus_1_3(const armnn_driver::hal_1_3::HalPolicy::Model& model,
                                                              armnn_driver::ArmnnDriver& driver,
-                                                             V1_3::ErrorStatus& prepareStatus)
+                                                             V1_3::ErrorStatus& prepareStatus,
+                                                             V1_3::Priority priority)
 {
     android::sp<PreparedModelCallback_1_3> cb(new PreparedModelCallback_1_3());
 
@@ -208,7 +209,7 @@
 
     driver.prepareModel_1_3(model,
                             V1_1::ExecutionPreference::LOW_POWER,
-                            V1_3::Priority::LOW,
+                            priority,
                             {},
                             emptyHandle1,
                             emptyHandle2,