Add missing function to fix the build.

Change-Id: I29f2a88803e904e40d82b1472abbcd0428a4f214