1 files changed, 5 insertions, 5 deletions
diff --git a/ArmnnDriver.hpp b/ArmnnDriver.hpp
index 2bf47eb3..fd5cfad0 100644
--- a/ArmnnDriver.hpp
+++ b/ArmnnDriver.hpp
@@ -9,18 +9,18 @@
 
 #include <log/log.h>
 
-#if defined(ARMNN_ANDROID_NN_V1_1) // Using ::android::hardware::neuralnetworks::V1_1.
+#if defined(ARMNN_ANDROID_NN_V1_1)
 
 #include "1.1/ArmnnDriver.hpp"
 
 namespace armnn_driver
 {
 
-class ArmnnDriver : public V1_1::ArmnnDriver
+class ArmnnDriver : public hal_1_1::ArmnnDriver
 {
 public:
     ArmnnDriver(DriverOptions options)
-        : V1_1::ArmnnDriver(std::move(options))
+        : hal_1_1::ArmnnDriver(std::move(options))
     {
         ALOGV("ArmnnDriver::ArmnnDriver()");
     }
@@ -36,11 +36,11 @@ public:
 namespace armnn_driver
 {
 
-class ArmnnDriver : public V1_0::ArmnnDriver
+class ArmnnDriver : public hal_1_0::ArmnnDriver
 {
 public:
     ArmnnDriver(DriverOptions options)
-        : V1_0::ArmnnDriver(std::move(options))
+        : hal_1_0::ArmnnDriver(std::move(options))
     {
         ALOGV("ArmnnDriver::ArmnnDriver()");
     }