athena: Fix switching windows with title-drag in split-view.

athena/wm/title_drag_controller.cc

Index: athena/wm/title_drag_controller.cc
diff --git a/athena/wm/title_drag_controller.cc b/athena/wm/title_drag_controller.cc
index 597f50a3c6de7e2b4521350696c18ce16e7a407e..766ea754aa49d6e6f6d11dedab421dcd488d6443 100644
--- a/athena/wm/title_drag_controller.cc
+++ b/athena/wm/title_drag_controller.cc
@@ -60,15 +60,15 @@ void TitleDragController::OnTransitionEnd(aura::Window* window, bool complete) {
   weak_ptr_.InvalidateWeakPtrs();
   if (!tracker_.Contains(window))
     window = NULL;
-  if (complete && window && wm::IsActiveWindow(window))
-    delegate_->OnTitleDragCompleted(window);
-  else
-    delegate_->OnTitleDragCanceled(window);
   shadow_.reset();
   if (window) {
     window->SetTransform(gfx::Transform());
     tracker_.Remove(window);
   }
+  if (complete && window && wm::IsActiveWindow(window))
+    delegate_->OnTitleDragCompleted(window);
+  else
+    delegate_->OnTitleDragCanceled(window);
 }
 
 void TitleDragController::OnGestureEvent(ui::GestureEvent* gesture) {