diff --git a/tests/AndroidTests/src/com/android/unit_tests/VpnTest.java b/tests/AndroidTests/src/com/android/unit_tests/VpnTest.java
index fd7861cb753b8..7dc13141e0c62 100755
--- a/tests/AndroidTests/src/com/android/unit_tests/VpnTest.java
+++ b/tests/AndroidTests/src/com/android/unit_tests/VpnTest.java
@@ -19,6 +19,7 @@ package com.android.unit_tests;
 import android.net.vpn.L2tpIpsecProfile;
 import android.net.vpn.VpnType;
 import android.test.AndroidTestCase;
+import android.test.suitebuilder.annotation.SmallTest;
 
 /**
  * Unit test class to test VPN api