FIX: roll back incorrect testing change of gradient order