diff --git a/include/media/IDrm.h b/include/media/IDrm.h
index 15d0a75..d630c40 100644
--- a/include/media/IDrm.h
+++ b/include/media/IDrm.h
@@ -17,6 +17,7 @@
 #include <binder/IInterface.h>
 #include <media/stagefright/foundation/ABase.h>
 #include <media/drm/DrmAPI.h>
+#include <media/IDrmClient.h>
 
 #ifndef ANDROID_IDRM_H_
 
@@ -106,6 +107,8 @@
                             Vector<uint8_t> const &signature,
                             bool &match) = 0;
 
+    virtual status_t setListener(const sp<IDrmClient>& listener) = 0;
+
 private:
     DISALLOW_EVIL_CONSTRUCTORS(IDrm);
 };