Updating unit test to use IDisplayController.