diff --git a/media/libaudiohal/StreamHalLocal.h b/media/libaudiohal/StreamHalLocal.h
index 5e6f41a..1df18cc 100644
--- a/media/libaudiohal/StreamHalLocal.h
+++ b/media/libaudiohal/StreamHalLocal.h
@@ -107,7 +107,7 @@
     virtual status_t supportsDrain(bool *supportsDrain);
 
     // Requests notification when data buffered by the driver/hardware has been played.
-    virtual status_t drain(audio_drain_type_t type);
+    virtual status_t drain(bool earlyNotify);
 
     // Notifies to the audio driver to flush the queued data.
     virtual status_t flush();