diff --git a/tools/scripts/java_tests_file.template b/tools/scripts/java_tests_file.template
index 7781a33c3..c6fa8736f 100644
--- a/tools/scripts/java_tests_file.template
+++ b/tools/scripts/java_tests_file.template
@@ -1,6 +1,6 @@
 package PACKAGE;
 
-import android.test.ActivityInstrumentationTestCase;
+import android.test.ActivityInstrumentationTestCase2;
 
 /**
  * This is a simple framework for a test of an Application.  See
@@ -12,7 +12,7 @@ import android.test.ActivityInstrumentationTestCase;
  * -e class PACKAGE.ACTIVITY_NAMETest \
  * PACKAGE.tests/android.test.InstrumentationTestRunner
  */
-public class ACTIVITY_NAMETest extends ActivityInstrumentationTestCase<ACTIVITY_NAME> {
+public class ACTIVITY_NAMETest extends ActivityInstrumentationTestCase2<ACTIVITY_NAME> {
 
     public ACTIVITY_NAMETest() {
         super("PACKAGE", ACTIVITY_NAME.class);