2 files changed, 5 insertions, 17 deletions

diff --git a/source/blender/editors/transform/transform.c b/source/blender/editors/transform/transform.c
index 6ff3826da09..6ce006a0f81 100644
--- a/source/blender/editors/transform/transform.c
+++ b/source/blender/editors/transform/transform.c
@@ -825,15 +825,11 @@ static void transform_event_xyz_constraint(TransInfo *t, short key_type, bool is
       }
     }
     else if (!edit_2d) {
-      if (cmode == axis) {
+      if (ELEM(cmode, '\0', axis)) {
         /* Successive presses on existing axis, cycle orientation modes. */
         t->orientation.index = (t->orientation.index + 1) % ARRAY_SIZE(t->orientation.types);
         initTransformOrientation(t->context, t, t->orientation.types[t->orientation.index]);
       }
-      else if (t->orientation.index != 1) {
-        t->orientation.index = 1;
-        initTransformOrientation(t->context, t, t->orientation.types[t->orientation.index]);
-      }
 
       if (t->orientation.index == 0) {
         stopConstraint(t);
diff --git a/source/blender/editors/transform/transform_generics.c b/source/blender/editors/transform/transform_generics.c
index 23f1295db3e..300186e2ecb 100644
--- a/source/blender/editors/transform/transform_generics.c
+++ b/source/blender/editors/transform/transform_generics.c
@@ -1719,21 +1719,13 @@ void initTransInfo(bContext *C, TransInfo *t, wmOperator *op, const wmEvent *eve
     }
 
     t->orientation.types[0] = orient_type_default;
+    t->orientation.types[1] = orient_type_constraint;
+    t->orientation.types[2] = orient_type_constraint != V3D_ORIENT_GLOBAL ? V3D_ORIENT_GLOBAL :
+                                                                            V3D_ORIENT_LOCAL;
     t->orientation.custom = custom_orientation;
 
-    /* To keep the old behavior logic to init contraint orientarions became this: */
-    t->orientation.types[1] = V3D_ORIENT_GLOBAL;
-    t->orientation.types[2] = orient_type_constraint != V3D_ORIENT_GLOBAL ?
-                                  orient_type_constraint :
-                                  V3D_ORIENT_LOCAL;
-
     if (t->con.mode & CON_APPLY) {
-      if (orient_type_constraint == V3D_ORIENT_GLOBAL) {
-        t->orientation.index = 1;
-      }
-      else {
-        t->orientation.index = 2;
-      }
+      t->orientation.index = 1;
     }
   }