Fixes output classes.