fix bug capturing screenshot and bugreport

Change-Id: I5fd623ccdcb6d64b993126e74232088ecacb3d0a
diff --git a/src/com/android/tradefed/testtype/UiAutomatorTest.java b/src/com/android/tradefed/testtype/UiAutomatorTest.java
index 03fb8a7..13d6b9f 100644
--- a/src/com/android/tradefed/testtype/UiAutomatorTest.java
+++ b/src/com/android/tradefed/testtype/UiAutomatorTest.java
@@ -262,7 +262,7 @@
                 mFailureAction == TestFailureAction.BUGREPORT_AND_SCREENSHOT) {
             InputStreamSource screenshot = null;
             try {
-                screenshot = getDevice().getScreenshot();
+                screenshot = device.getScreenshot();
                 listener.testLog(prefix + "_screenshot", LogDataType.PNG, screenshot);
             } catch (DeviceNotAvailableException e) {
                 CLog.e(e);
@@ -276,7 +276,7 @@
         if (mFailureAction == TestFailureAction.BUGREPORT ||
                 mFailureAction == TestFailureAction.BUGREPORT_AND_SCREENSHOT) {
             InputStreamSource data = null;
-            data = getDevice().getBugreport();
+            data = device.getBugreport();
             listener.testLog(prefix + "_bugreport", LogDataType.BUGREPORT, data);
             if (data != null) {
                 data.cancel();