diff --git a/packages/auth/src/providers/cognito/types/models.ts b/packages/auth/src/providers/cognito/types/models.ts
index a65d738127d..1b113ef1720 100644
--- a/packages/auth/src/providers/cognito/types/models.ts
+++ b/packages/auth/src/providers/cognito/types/models.ts
@@ -10,7 +10,6 @@ import {
 	AWSAuthUser,
 	AuthCodeDeliveryDetails,
 	AuthDevice,
-	AuthUserAttribute,
 } from '../../../types';
 import { AuthProvider } from '../../../types/inputs';
 
@@ -89,11 +88,15 @@ export type AutoSignInEventData =
 	| {
 			event: 'autoSignIn';
 	  };
+
+type DeviceAttributeKey = 'device_status' | 'device_name' | 'last_ip_used';
+
 /**
  * Holds the device specific information along with it's id and name.
  */
 export type AWSAuthDevice = AuthDevice & {
-	attributes: AuthUserAttribute<UserAttributeKey>;
+	attributes: Partial<Record<DeviceAttributeKey, string>> &
+		Record<string, string>;
 	createDate?: Date;
 	lastAuthenticatedDate?: Date;
 	lastModifiedDate?: Date;