diff --git a/custom_components/stockpile/number.py b/custom_components/stockpile/number.py
index 4bbedb8..c0a44f0 100644
--- a/custom_components/stockpile/number.py
+++ b/custom_components/stockpile/number.py
@@ -18,7 +18,7 @@ async def async_setup_entry(
     """Set up number platform."""
     async_add_entities([StockPileNumber(entry)])
 
-class StockPileNumber(NumberEntity):
+class StockPileNumber(RestoreNumber, NumberEntity):
     """StockPile number class."""
 
     def __init__(self, entry: ConfigEntry) -> None: