diff --git a/src/Traits/HandlesPayment.php b/src/Traits/HandlesPayment.php
index 868e590..d8d83ca 100644
--- a/src/Traits/HandlesPayment.php
+++ b/src/Traits/HandlesPayment.php
@@ -26,8 +26,7 @@ public function pay(int|float $orderValue, ?string $notes = null): void
             $remainingOrderValue = $orderValue;
 
             /**
-             * @var \Illuminate\Support\Collection<TKey, \HPWebdeveloper\LaravelPayPocket\Models
-             * |\HPWebdeveloper\LaravelPayPocket\Traits\BalanceOperation>  $walletsInOrder
+             * @var \Illuminate\Support\Collection<TKey, \HPWebdeveloper\LaravelPayPocket\Models\Wallet>
              */
             $walletsInOrder = $this->wallets()->whereIn('type', $this->walletsInOrder())->get();