Fix for old class type hint