Fix to generalization