Fix type hints for the compiler