Fix type hint