From 2ab18544c309eacc37ad89885662cf4fb34f0423 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?dni=20=E2=9A=A1?= <office@dnilabs.com>
Date: Thu, 24 Aug 2023 08:26:34 +0200
Subject: [PATCH] [BUG] use WalletType for get_wallet_for_key fn (#1896)

---
 lnbits/core/crud.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lnbits/core/crud.py b/lnbits/core/crud.py
index 3656df9e7..dc8991f61 100644
--- a/lnbits/core/crud.py
+++ b/lnbits/core/crud.py
@@ -281,7 +281,9 @@ async def get_wallet(
 
 
 async def get_wallet_for_key(
-    key: str, key_type: str = "invoice", conn: Optional[Connection] = None
+    key: str,
+    key_type: WalletType = WalletType.invoice,
+    conn: Optional[Connection] = None,
 ) -> Optional[Wallet]:
     row = await (conn or db).fetchone(
         """