diff --git a/lnbits/app.py b/lnbits/app.py
index 01d93f61d..0ee7f4f5b 100644
--- a/lnbits/app.py
+++ b/lnbits/app.py
@@ -19,7 +19,7 @@ from starlette.middleware.sessions import SessionMiddleware
 from lnbits.core.crud import (
     get_dbversions,
     get_installed_extensions,
-    update_installed_extension,
+    update_installed_extension_state,
 )
 from lnbits.core.extensions.helpers import version_parse
 from lnbits.core.helpers import migrate_extension_database
@@ -239,8 +239,7 @@ async def check_installed_extensions(app: FastAPI):
         except Exception as e:
             logger.warning(e)
             settings.deactivate_extension_paths(ext.id)
-            ext.active = False
-            await update_installed_extension(ext)
+            await update_installed_extension_state(ext_id=ext.id, active=False)
             logger.warning(
                 f"Failed to re-install extension: {ext.id} ({ext.installed_version})"
             )
@@ -309,8 +308,7 @@ async def check_installed_extension_files(ext: InstallableExtension) -> bool:
 
 
 async def restore_installed_extension(app: FastAPI, ext: InstallableExtension):
-    ext.active = True
-    await update_installed_extension(ext)
+    await update_installed_extension_state(ext_id=ext.id, active=True)
 
     extension = Extension.from_installable_ext(ext)
     register_ext_routes(app, extension)
diff --git a/lnbits/core/models.py b/lnbits/core/models.py
index 3d0001c4b..56ed27e1b 100644
--- a/lnbits/core/models.py
+++ b/lnbits/core/models.py
@@ -41,7 +41,6 @@ class Wallet(BaseModel):
     name: str
     adminkey: str
     inkey: str
-    balance_msat: int
     currency: Optional[str]
     deleted: bool = False
     created_at: Optional[int] = None