Fix invalid type hint