Fix type

Change-Id: Ib44a61073269540788967bbe658cc94678652cf1