diff --git a/include/aidge/learning/optimizer/SGD.hpp b/include/aidge/learning/optimizer/SGD.hpp
index ce475463fe5ac385e9eac129a9cc598c0c250c8a..2ce657282496de90e5410e5e128a715a474d79bc 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"