diff --git a/VoldNativeService.h b/VoldNativeService.h
index 744ff84..cc4dd1b 100644
--- a/VoldNativeService.h
+++ b/VoldNativeService.h
@@ -53,7 +53,7 @@
     binder::Status forgetPartition(const std::string& partGuid, const std::string& fsUuid);
 
     binder::Status mount(const std::string& volId, int32_t mountFlags, int32_t mountUserId,
-                         android::base::unique_fd* _aidl_return);
+                         const android::sp<android::os::IVoldMountCallback>& callback);
     binder::Status unmount(const std::string& volId);
     binder::Status format(const std::string& volId, const std::string& fsType);
     binder::Status benchmark(const std::string& volId,