diff --git a/Test/Unit/Model/Api/GuestAdyenStateDataTest.php b/Test/Unit/Model/Api/GuestAdyenStateDataTest.php
index 29809c43d..5f6f42db1 100644
--- a/Test/Unit/Model/Api/GuestAdyenStateDataTest.php
+++ b/Test/Unit/Model/Api/GuestAdyenStateDataTest.php
@@ -57,11 +57,8 @@ public function testSaveSuccessful()
 
         $this->quoteIdMaskFactoryMock->method('create')->willReturn($this->quoteIdMaskMock);
         $this->stateDataHelperMock->expects($this->once())->method('saveStateData')->willReturn($stateDataMock);
-        //should return int
-
 
         $this->guestAdyenStateDataModel->save($stateData, $cartId);
-//        $this->assertIsInt($result);
     }
 
     public function testRemoveSuccessful()
diff --git a/Test/Unit/Model/Resolver/GetAdyenPaymentMethodsBalanceTest.php b/Test/Unit/Model/Resolver/GetAdyenPaymentMethodsBalanceTest.php
index bdbcefcad..01f905318 100644
--- a/Test/Unit/Model/Resolver/GetAdyenPaymentMethodsBalanceTest.php
+++ b/Test/Unit/Model/Resolver/GetAdyenPaymentMethodsBalanceTest.php
@@ -71,28 +71,6 @@ public function testWithValidPayloadArgument()
 
         $this->assertEquals(['balanceResponse' => $expectedBalanceResponse], $result);
     }
-
-//    public function testWithAdyensAPIReturningAnError()
-//    {
-//        $payload = '{"some":"data"}';
-//        $args = ['payload' => $payload];
-//
-//        $this->balanceMock->expects($this->once())
-//            ->method('getBalance')
-//            ->with($payload)
-//            ->willThrowException(new Exception('API Error'));
-//
-//        $this->expectException(GraphQlAdyenException::class);
-//        $this->expectExceptionMessage('An error occurred while fetching the payment method balance.');
-//
-//        $fieldMock = $this->createMock(\Magento\Framework\GraphQl\Query\Fields::class);
-//        $contextMock = $this->createMock(\Magento\Framework\GraphQl\Config\Element\Field::class);
-//        $resolveInfoMock = $this->createMock(ResolveInfo::class);
-//
-//        $result = $this->getAdyenPaymentMethodsBalance->resolve($fieldMock, $contextMock, $resolveInfoMock, [], $args);
-//    }
-
-
 }