Fix error in code

Change-Id: Ic811aab0c91c59da65eb11be1cadee34bcb0fad0