Fixed a type hint problem