Index: ash/display/window_tree_host_manager_unittest.cc |
diff --git a/ash/display/window_tree_host_manager_unittest.cc b/ash/display/window_tree_host_manager_unittest.cc |
index b64ebc244fc967068dcbe6f6be97210b2f329a25..b2a63a736dbeee102abf05d2b19bf8503df67e1b 100644 |
--- a/ash/display/window_tree_host_manager_unittest.cc |
+++ b/ash/display/window_tree_host_manager_unittest.cc |
@@ -9,8 +9,8 @@ |
#include "ash/aura/wm_window_aura.h" |
#include "ash/common/ash_switches.h" |
#include "ash/common/material_design/material_design_controller.h" |
-#include "ash/common/shelf/shelf.h" |
#include "ash/common/shelf/shelf_widget.h" |
+#include "ash/common/shelf/wm_shelf.h" |
#include "ash/common/wm/window_state.h" |
#include "ash/common/wm/wm_event.h" |
#include "ash/common/wm/wm_screen_util.h" |
@@ -818,7 +818,7 @@ TEST_P(WindowTreeHostManagerTest, SwapPrimaryById) { |
window_tree_host_manager->GetRootWindowForDisplayId( |
secondary_display.id()); |
aura::Window* shelf_window = |
- Shelf::ForPrimaryDisplay()->shelf_widget()->GetNativeView(); |
+ GetPrimaryShelf()->shelf_widget()->GetNativeView(); |
EXPECT_TRUE(primary_root->Contains(shelf_window)); |
EXPECT_FALSE(secondary_root->Contains(shelf_window)); |
EXPECT_NE(primary_root, secondary_root); |