diff --git a/lib/main/exceptions/remote/authentication_exception.dart b/lib/main/exceptions/remote/authentication_exception.dart
index 9e2f056c7..55f201b73 100644
--- a/lib/main/exceptions/remote/authentication_exception.dart
+++ b/lib/main/exceptions/remote/authentication_exception.dart
@@ -1,7 +1,7 @@
 import 'package:tmail_ui_user/main/exceptions/remote/remote_exception.dart';
 
 class AuthenticationException extends RemoteException {
-  const AuthenticationException({super.message});
+  const AuthenticationException({super.code, super.message});
 
   @override
   String get exceptionName => 'AuthenticationException';
@@ -19,6 +19,7 @@ class RefreshTokenFailedException extends AuthenticationException {
     int code = 400,
     String? message,
   }) : super(
+          code: code,
           message: message ??
               'Refresh token failed with status $code. Session invalid or revoked.',
         );
diff --git a/test/features/base/base_controller_test.dart b/test/features/base/base_controller_test.dart
index ee185b8d5..9f7ce74be 100644
--- a/test/features/base/base_controller_test.dart
+++ b/test/features/base/base_controller_test.dart
@@ -57,6 +57,8 @@ class MockBaseController extends BaseController {
 }
 
 class SomeOtherException extends RemoteException {
+  const SomeOtherException();
+
   @override
   String get exceptionName => 'SomeOtherException';
 }
@@ -145,7 +147,7 @@ void main() {
     });
 
     test('should return false when exception is SomeOtherException', () {
-      expect(mockBaseController.validateUrgentException(SomeOtherException()), isFalse);
+      expect(mockBaseController.validateUrgentException(const SomeOtherException()), isFalse);
     });
 
     test('should return false when exception is null', () {
@@ -204,7 +206,7 @@ void main() {
 
     test('handleErrorViewState should called when error is SomeOtherException', () {
       // arrange
-      final error = SomeOtherException();
+      const error = SomeOtherException();
       final stackTrace = StackTrace.current;
 
       // act