Fix type hint

Change-Id: If29388a299781c13a36fc26a15413094310db210