Fix issue with incorrect type hinting