diff --git a/pytimeloop/timeloopfe/v4/arch.py b/pytimeloop/timeloopfe/v4/arch.py
index 04bde53..e954a60 100644
--- a/pytimeloop/timeloopfe/v4/arch.py
+++ b/pytimeloop/timeloopfe/v4/arch.py
@@ -383,6 +383,8 @@ def declare_attrs(cls, *args, **kwargs):
     def __init__(self, *args, **kwargs):
         super().__init__(self, *args, **kwargs)
         self.attributes: StorageAttributes = self["attributes"]
+        if self.attributes.depth is None:
+            print(f'WARNING: "depth" is not set for storage element {self.name}')
 
 
 class Compute(Component):
@@ -587,8 +589,6 @@ def __init__(self, *args, **kwargs):
         ]
         self.decompression_supported: Union[str, bool] = self["decompression_supported"]
         self.compression_supported: Union[str, bool] = self["compression_supported"]
-        if self.depth is None:
-            print(f'WARNING: "depth" is not set for storage element {self.name}')
 
 
 class Nothing(Component):