Bug fix assign subclass