Fixing registering small classes.