Maybe before removing that classes assert if it is there? (It should be if GCN worked correctly).