Trying to fix subclass issue