diff --git a/modules/order/tests/src/Kernel/Entity/OrderTest.php b/modules/order/tests/src/Kernel/Entity/OrderTest.php
index 62b15d8a8..8e05fb4ac 100644
--- a/modules/order/tests/src/Kernel/Entity/OrderTest.php
+++ b/modules/order/tests/src/Kernel/Entity/OrderTest.php
@@ -300,6 +300,7 @@ public function testOrder() {
     $order = Order::load($order->id());
     $this->assertEquals(new Price('25.00', 'USD'), $order->getTotalPaid());
     $this->assertEquals(new Price('16.00', 'USD'), $order->getBalance());
+    $payment->setState('partially_refunded')->save();
     $payment->setRefundedAmount(new Price('5.00', 'USD'))->save();
     $order = Order::load($order->id());
     $this->assertEquals(new Price('20.00', 'USD'), $order->getTotalPaid());