Fix class hinting bug