Fix function signature

Change-Id: Ia3cd228009792d2e5c2e3b74e740341bb0088867