Change to fix class decorators.