Fixed a bug with updating subclasses.