Fixed bug with type hints