Add missing function

Change-Id: I4f261069398ff1a4f475886966a3170c835b245f