From 871d3b66fb5f3d8484456f4da414f56411989fad Mon Sep 17 00:00:00 2001
From: Otto Winter <otto@otto-winter.com>
Date: Mon, 4 Oct 2021 16:15:25 +0200
Subject: [PATCH] Fix restoring globals (#2442)

---
 esphome/cpp_generator.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/esphome/cpp_generator.py b/esphome/cpp_generator.py
index cf357f2814..937b6cceb4 100644
--- a/esphome/cpp_generator.py
+++ b/esphome/cpp_generator.py
@@ -638,7 +638,7 @@ async def process_lambda(
     :param return_type: The return type of the lambda.
     :return: The generated lambda expression.
     """
-    from esphome.components.globals import GlobalsComponent
+    from esphome.components.globals import GlobalsComponent, RestoringGlobalsComponent
 
     if value is None:
         return
@@ -648,7 +648,10 @@ async def process_lambda(
         if (
             full_id is not None
             and isinstance(full_id.type, MockObjClass)
-            and full_id.type.inherits_from(GlobalsComponent)
+            and (
+                full_id.type.inherits_from(GlobalsComponent)
+                or full_id.type.inherits_from(RestoringGlobalsComponent)
+            )
         ):
             parts[i * 3 + 1] = var.value()
             continue