1 files changed, 2 insertions, 11 deletions
diff --git a/src/com/android/tradefed/targetprep/AddWhitelistPackage.java b/src/com/android/tradefed/targetprep/AddWhitelistPackage.java
index 0bd0109..92e628c 100644
--- a/src/com/android/tradefed/targetprep/AddWhitelistPackage.java
+++ b/src/com/android/tradefed/targetprep/AddWhitelistPackage.java
@@ -29,14 +29,6 @@ import java.util.List;
  */
 @OptionClass(alias = "add-whitelist-package")
 public class AddWhitelistPackage extends BaseTargetPreparer implements ITargetCleaner {
-
-    @Option(
-            name = "whitelist-package-time",
-            description = "Duration to put package in whitelist",
-            isTimeVal = true
-    )
-    private long mDurationMillis = 10000L;
-
     @Option(
             name = "whitelist-package-name",
             description = "Name of package to put in whitelist"
@@ -48,8 +40,7 @@ public class AddWhitelistPackage extends BaseTargetPreparer implements ITargetCl
             throws TargetSetupError, BuildError, DeviceNotAvailableException {
         for (String pkg : mPackages) {
             device.executeShellCommand(
-                    String.format("dumpsys deviceidle tempwhitelist -d %d %s", mDurationMillis,
-                            pkg));
+                    String.format("dumpsys deviceidle whitelist +%s", pkg));
         }
     }
 
@@ -58,7 +49,7 @@ public class AddWhitelistPackage extends BaseTargetPreparer implements ITargetCl
             throws DeviceNotAvailableException {
         for (String pkg : mPackages) {
             device.executeShellCommand(
-                    String.format("dumpsys deviceidle tempwhitelist -r %s", pkg));
+                    String.format("dumpsys deviceidle whitelist -%s", pkg));
         }
     }
 }