Fix prototype

Change-Id: I251ba2ab9dec3cbb74b506cf173d8759805d5a1c