Fix feature axis comparison for both original variable addition and subsequent references;