Fix WifiStateMachineTest

Bug: 65599523
Bug: 65550505
Test: compile, unit tests
Change-Id: Ie9092dae5b07ac14889545c41b2ca212eca97294
diff --git a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
index 74c6b79..07840fd 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 @@
     @Mock WrongPasswordNotifier mWrongPasswordNotifier;
     @Mock Clock mClock;
     @Mock ScanDetailCache mScanDetailCache;
-    @Mock WifiDiagnostics mWifiDiagnostics;
+    @Mock BaseWifiDiagnostics mWifiDiagnostics;
 
     public WifiStateMachineTest() throws Exception {
     }