Fix type hinting