diff --git a/runtime/Callbacks.cpp b/runtime/Callbacks.cpp
index 4fd4f0b..44c97fa 100644
--- a/runtime/Callbacks.cpp
+++ b/runtime/Callbacks.cpp
@@ -19,10 +19,11 @@
 #include "Callbacks.h"
 
 #include <android-base/logging.h>
-
 #include <limits>
 
-namespace android::hardware::neuralnetworks::V1_2::implementation {
+namespace android::nn {
+
+using namespace hal;
 
 constexpr Timing kNoTiming = {.timeOnDevice = std::numeric_limits<uint64_t>::max(),
                               .timeInDriver = std::numeric_limits<uint64_t>::max()};
@@ -206,4 +207,4 @@
     mCondition.notify_all();
 }
 
-}  // namespace android::hardware::neuralnetworks::V1_2::implementation
+}  // namespace android::nn