Fixed incorrect class type hint