Fixes the incorrect type hinting