diff --git a/include/aidge/learning/optimizer/SGD.hpp b/include/aidge/learning/optimizer/SGD.hpp
index 2f77281500b5f92c52146c60652f94530d12574c..4a2050cb6528883c37e28d1c5a85a6bc5d74053a 100644
--- a/include/aidge/learning/optimizer/SGD.hpp
+++ b/include/aidge/learning/optimizer/SGD.hpp
@@ -17,6 +17,7 @@
 #include <vector>
 
 #include "aidge/data/Tensor.hpp"
+#include "aidge/backend/cpu/data/TensorImpl.hpp"
 #include "aidge/learning/optimizer/Optimizer.hpp"
 #include "aidge/utils/StaticAttributes.hpp"
 #include "aidge/utils/Registrar.hpp"