diff --git a/IdentityCore/tests/util/MSIDTestCacheDataSource.h b/IdentityCore/tests/util/MSIDTestCacheDataSource.h
index ccf5e15f9..28f2c898b 100644
--- a/IdentityCore/tests/util/MSIDTestCacheDataSource.h
+++ b/IdentityCore/tests/util/MSIDTestCacheDataSource.h
@@ -23,7 +23,7 @@
 
 #import <Foundation/Foundation.h>
 #import "MSIDExtendedTokenCacheDataSource.h"
-#import "MSIDBaseToken.h"
+#import "MSIDCredentialType.h"
 #import "MSIDCacheItemSerializing.h"
 
 @interface MSIDTestCacheDataSource : NSObject <MSIDExtendedTokenCacheDataSource>