diff --git a/VolumeManager.h b/VolumeManager.h
index 7dc69f3..2751ad5 100644
--- a/VolumeManager.h
+++ b/VolumeManager.h
@@ -36,6 +36,8 @@
 #include <sysutils/SocketListener.h>
 #include <sysutils/NetlinkEvent.h>
 
+#include "android/os/IVoldListener.h"
+
 #include "model/Disk.h"
 #include "model/VolumeBase.h"
 
@@ -96,6 +98,9 @@
     std::mutex& getLock() { return mLock; }
     std::mutex& getCryptLock() { return mCryptLock; }
 
+    void setListener(android::sp<android::os::IVoldListener> listener) { mListener = listener; }
+    android::sp<android::os::IVoldListener> getListener() { return mListener; }
+
     int start();
     int stop();
 
@@ -221,6 +226,8 @@
     std::mutex mLock;
     std::mutex mCryptLock;
 
+    android::sp<android::os::IVoldListener> mListener;
+
     std::list<std::shared_ptr<DiskSource>> mDiskSources;
     std::list<std::shared_ptr<android::vold::Disk>> mDisks;
     std::list<std::shared_ptr<android::vold::VolumeBase>> mObbVolumes;