diff --git a/src/test/java/io/supertokens/storage/postgresql/test/multitenancy/StorageLayerTest.java b/src/test/java/io/supertokens/storage/postgresql/test/multitenancy/StorageLayerTest.java
index 77204865..b75bf745 100644
--- a/src/test/java/io/supertokens/storage/postgresql/test/multitenancy/StorageLayerTest.java
+++ b/src/test/java/io/supertokens/storage/postgresql/test/multitenancy/StorageLayerTest.java
@@ -174,11 +174,11 @@ public void storageInstanceIsReusedAcrossTenants()
                 StorageLayer.getStorage(new TenantIdentifier(null, null, null), process.getProcess()));
 
         Assert.assertEquals(
-                Config.getConfig(new TenantIdentifier(null, null, null), process.getProcess()).getAccessTokenValidity(),
+                Config.getConfig(new TenantIdentifier(null, null, null), process.getProcess()).getAccessTokenValidityInMillis(),
                 (long) 3600 * 1000);
 
         Assert.assertEquals(Config.getConfig(new TenantIdentifier(null, "abc", null), process.getProcess())
-                        .getAccessTokenValidity(),
+                        .getAccessTokenValidityInMillis(),
                 (long) 3601 * 1000);
 
         Assert.assertEquals(
@@ -238,11 +238,11 @@ public void storageInstanceIsReusedAcrossTenantsComplex()
                 StorageLayer.getStorage(new TenantIdentifier(null, null, null), process.getProcess()));
 
         Assert.assertEquals(
-                Config.getConfig(new TenantIdentifier(null, null, null), process.getProcess()).getAccessTokenValidity(),
+                Config.getConfig(new TenantIdentifier(null, null, null), process.getProcess()).getAccessTokenValidityInMillis(),
                 (long) 3600 * 1000);
 
         Assert.assertEquals(Config.getConfig(new TenantIdentifier(null, "abc", null), process.getProcess())
-                        .getAccessTokenValidity(),
+                        .getAccessTokenValidityInMillis(),
                 (long) 3601 * 1000);
 
         Assert.assertEquals(
@@ -453,11 +453,11 @@ public void newStorageIsNotCreatedWhenSameTenantIsAdded()
                 existingStorage);
 
         Assert.assertEquals(
-                Config.getConfig(new TenantIdentifier(null, null, null), process.getProcess()).getAccessTokenValidity(),
+                Config.getConfig(new TenantIdentifier(null, null, null), process.getProcess()).getAccessTokenValidityInMillis(),
                 (long) 3600 * 1000);
 
         Assert.assertEquals(Config.getConfig(new TenantIdentifier(null, "abc", null), process.getProcess())
-                        .getAccessTokenValidity(),
+                        .getAccessTokenValidityInMillis(),
                 (long) 3601 * 1000);
 
         Assert.assertEquals(