fixing some type hints