Fixes type hinting