Add type hint

Change-Id: If9ad708f5dcb792cfcc74d87e2f82af4f9a1c31d