diff --git a/src/test/java/com/safetypin/authentication/service/TokenExpirationTest.java b/src/test/java/com/safetypin/authentication/service/TokenExpirationTest.java
index 099681cb485a60e35e7405a7b08920f5d008b5b3..0b9ac09bf42c156f70fc5a11c813333207aa69c3 100644
--- a/src/test/java/com/safetypin/authentication/service/TokenExpirationTest.java
+++ b/src/test/java/com/safetypin/authentication/service/TokenExpirationTest.java
@@ -1,6 +1,5 @@
 package com.safetypin.authentication.service;
 
-import com.safetypin.authentication.dto.UserResponse;
 import com.safetypin.authentication.exception.InvalidCredentialsException;
 import com.safetypin.authentication.model.User;
 import com.safetypin.authentication.repository.UserRepository;
@@ -20,9 +19,7 @@ import java.util.UUID;
 import static org.junit.jupiter.api.Assertions.*;
 import static org.mockito.Mockito.when;
 
-public class TokenExpirationTest {
-
-
+class TokenExpirationTest {
     @Mock
     private UserRepository userRepository;
 
@@ -79,7 +76,7 @@ public class TokenExpirationTest {
         }
 
         // This method simulates the token expiration check portion of getUserFromJwtToken
-        public UserResponse testTokenExpiration(UUID userId) {
+        public void testTokenExpiration(UUID userId) {
             try {
                 // Mock a Claims object with an expired date
                 Claims claims = Jwts.claims()
@@ -99,7 +96,7 @@ public class TokenExpirationTest {
                 if (user.isEmpty()) {
                     throw new InvalidCredentialsException("User not found");
                 }
-                return user.get().generateUserResponse();
+                user.get().generateUserResponse();
             } catch (JwtException | IllegalArgumentException e) {
                 throw new InvalidCredentialsException("Invalid token");
             }