Fixed a type hinting error