diff --git a/Tests/Unit/inc/classes/ImagifyUser/getError.php b/Tests/Unit/inc/classes/ImagifyUser/getError.php
index 565aebe4..035012cf 100644
--- a/Tests/Unit/inc/classes/ImagifyUser/getError.php
+++ b/Tests/Unit/inc/classes/ImagifyUser/getError.php
@@ -31,7 +31,7 @@ public function testShouldReturnFalseWhenFetchedUserData() {
 			'is_monthly'                   => true,
 		];
 
-		Functions\when( 'get_transient' )->justReturn( false );
+		Functions\when( 'imagify_get_cached_user' )->justReturn( false );
 		Functions\when( 'get_imagify_user' )->justReturn( $userData );
 
 		$this->assertFalse( ( new User() )->get_error() );
@@ -55,7 +55,7 @@ public function testShouldReturnFromCachedUserDataIfAvailable() {
 			'is_monthly'                   => true,
 		];
 
-		Functions\when( 'get_transient' )->justReturn( $userData );
+		Functions\when( 'imagify_get_cached_user' )->justReturn( $userData );
 		Functions\expect( 'get_imagify_user' )->never();
 
 		$this->assertSame( 'imagify@example.com', ( new User() )->email );
@@ -67,7 +67,7 @@ public function testShouldReturnFromCachedUserDataIfAvailable() {
 	public function testShouldReturnErrorWhenCouldNotFetchUserData() {
 		$wp_error = new WP_Error( 'error_id', 'Error Message' );
 
-		Functions\when( 'get_transient' )->justReturn( false );
+		Functions\when( 'imagify_get_cached_user' )->justReturn( false );
 		Functions\when( 'get_imagify_user' )->justReturn( $wp_error );
 
 		$this->assertSame( $wp_error, ( new User() )->get_error() );
diff --git a/Tests/Unit/inc/classes/ImagifyUser/getPercentConsumedQuota.php b/Tests/Unit/inc/classes/ImagifyUser/getPercentConsumedQuota.php
index d56b081c..a981d338 100644
--- a/Tests/Unit/inc/classes/ImagifyUser/getPercentConsumedQuota.php
+++ b/Tests/Unit/inc/classes/ImagifyUser/getPercentConsumedQuota.php
@@ -23,7 +23,7 @@ class Test_GetPercentConsumedQuota extends TestCase {
 	public function testShouldReturnZeroWhenCouldNotFetchUserData() {
 		$wp_error = new WP_Error( 'error_id', 'Error Message' );
 
-		Functions\when( 'get_transient' )->justReturn( false );
+		Functions\when( 'imagify_get_cached_user' )->justReturn( false );
 		Functions\when( 'get_imagify_user' )->justReturn( $wp_error );
 		Functions\expect( 'imagify_round_half_five' )->never();
 
@@ -48,7 +48,7 @@ public function testShouldReturnQuotaWhenFetchedUserData() {
 			'is_monthly'                   => true,
 		];
 
-		Functions\when( 'get_transient' )->justReturn( false );
+		Functions\when( 'imagify_get_cached_user' )->justReturn( false );
 		Functions\when( 'get_imagify_user' )->justReturn( $userData );
 		Functions\expect( 'imagify_round_half_five' )
 			->twice()
@@ -72,7 +72,7 @@ public function testShouldReturnQuotaWhenFetchedUserData() {
 
 		$userData->consumed_current_month_quota = 500; // Current consumed quota 50%.
 
-		Functions\when( 'get_transient' )->justReturn( $userData );
+		Functions\when( 'get_imagify_user' )->justReturn( $userData );
 
 		$imagify_data_mock->shouldReceive( 'get' )
 			->never();
diff --git a/Tests/Unit/inc/classes/ImagifyUser/isOverQuota.php b/Tests/Unit/inc/classes/ImagifyUser/isOverQuota.php
index 0942850e..8fbd48b1 100644
--- a/Tests/Unit/inc/classes/ImagifyUser/isOverQuota.php
+++ b/Tests/Unit/inc/classes/ImagifyUser/isOverQuota.php
@@ -22,7 +22,7 @@ class Test_IsOverQuota extends TestCase {
 	public function testShouldReturnFalseWhenCouldNotFetchUserData() {
 		$wp_error = new WP_Error( 'error_id', 'Error Message' );
 
-		Functions\when( 'get_transient' )->justReturn( false );
+		Functions\when( 'imagify_get_cached_user' )->justReturn( false );
 		Functions\when( 'get_imagify_user' )->justReturn( $wp_error );
 
 		$this->assertFalse( ( new User() )->is_over_quota() );
@@ -46,7 +46,7 @@ public function testShouldReturnFalseWhenPaidAccount() {
 			'is_monthly'                   => true,
 		];
 
-		Functions\when( 'get_transient' )->justReturn( false );
+		Functions\when( 'imagify_get_cached_user' )->justReturn( false );
 		Functions\when( 'get_imagify_user' )->justReturn( $userData );
 
 		$this->assertFalse( ( new User() )->is_over_quota() );
@@ -99,7 +99,7 @@ public function testShouldReturnTrueWhenFreeOverQuota() {
 	}
 
 	private function createMocks( $userData, $dataPreviousQuotaPercent ) {
-		Functions\when( 'get_transient' )->justReturn( false );
+		Functions\when( 'imagify_get_cached_user' )->justReturn( false );
 		Functions\when( 'get_imagify_user' )->justReturn( $userData );
 		Functions\expect( 'imagify_round_half_five' )
 			->once()