Fix doc type hint