1 files changed, 1 insertions, 1 deletions
diff --git a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
index 74c6b79ee..07840fdac 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
@@ -359,7 +359,7 @@ public class WifiStateMachineTest {
     @Mock WrongPasswordNotifier mWrongPasswordNotifier;
     @Mock Clock mClock;
     @Mock ScanDetailCache mScanDetailCache;
-    @Mock WifiDiagnostics mWifiDiagnostics;
+    @Mock BaseWifiDiagnostics mWifiDiagnostics;
 
     public WifiStateMachineTest() throws Exception {
     }