diff --git a/app/Endpoint/Api/Http/Controllers/Account/WalletController.php b/app/Endpoint/Api/Http/Controllers/Account/WalletController.php
index 6b1060d8..ec18965c 100644
--- a/app/Endpoint/Api/Http/Controllers/Account/WalletController.php
+++ b/app/Endpoint/Api/Http/Controllers/Account/WalletController.php
@@ -126,22 +126,20 @@ class WalletController extends Controller
                                 ->lockForUpdate()
                                 ->first();
 
-                if ($payLog === null || !in_array($payLog->pay_way, [PayLog::PAY_WAY_WALLET, PayLog::PAY_WAY_BALANCE])) {
+                if ($payLog === null) {
                     throw new InvalidPaySerialNumberException();
                 }
 
-                switch ($payLog->pay_way) {
-                    case PayLog::PAY_WAY_WALLET:
-                        if ($user->wallet->is_frozen) {
-                            throw new WalletFrozenException();
-                        }
-                        break;
-
-                    case PayLog::PAY_WAY_BALANCE:
-                        if ($user->balance->is_frozen) {
-                            throw new BalanceFrozenException();
-                        }
-                        break;
+                if ($payLog->isBalance()) {
+                    if ($user->balance->is_frozen) {
+                        throw new BalanceFrozenException();
+                    }
+                } elseif ($payLog->isWallet()) {
+                    if ($user->wallet->is_frozen) {
+                        throw new WalletFrozenException();
+                    }
+                } else {
+                    throw new InvalidPaySerialNumberException();
                 }
 
                 $payable = $payLog->payable;