Fix type hint

Change-Id: I4f803ccf1bd03dd65adf7bedcd61937b2e89bf13