fixed wrong type hint