diff --git a/deevirt_schema/api/stats.py b/deevirt_schema/api/stats.py
index eeb58b5..ce1ac2e 100644
--- a/deevirt_schema/api/stats.py
+++ b/deevirt_schema/api/stats.py
@@ -1,13 +1,13 @@
 import datetime
 from typing import Optional
 from pydantic import BaseModel, ConfigDict, Field
-from ..libvirt.dom.stats import Stats
+from ..libvirt.dom.stats import Stats, Memory
 
 class MemorySwap(BaseModel):
     swp_in:Optional[int] = Field(default=None)
     swp_out:Optional[int] = Field(default=None)
 
-class Memory(BaseModel):
+class Memory(Memory):
     swap:MemorySwap = Field()
 
 class Stats(Stats):