fix type hint

Change-Id: Ic5110d923b6a2e384210fc68426c111eb5bb740e