diff --git a/unit_tests/operator/Test_HeavisideImpl.cpp b/unit_tests/operator/Test_HeavisideImpl.cpp
index 9241f0d37619d097bf8a3e81945a18e08ce86017..d3ed38263f8fe82e5b5b5fbcc03a8f7dcd761c73 100644
--- a/unit_tests/operator/Test_HeavisideImpl.cpp
+++ b/unit_tests/operator/Test_HeavisideImpl.cpp
@@ -11,23 +11,22 @@
 
 #include "aidge/backend/cpu/operator/HeavisideImpl_kernels.hpp"
 
-#include <aidge/operator/Atan.hpp>
-#include <aidge/operator/Memorize.hpp>
-#include <aidge/operator/Mul.hpp>
-#include <aidge/operator/Producer.hpp>
-#include <aidge/utils/Types.h>
 #include <memory>
+#include <cmath>
 #include <cstdlib>
 #include <random>
 
 #include <catch2/catch_test_macros.hpp>
 
-#include "aidge/data/Tensor.hpp"
 #include "aidge/backend/cpu/operator/HeavisideImpl.hpp"
+#include "aidge/data/Tensor.hpp"
 #include "aidge/graph/Node.hpp"
+#include "aidge/operator/Atan.hpp"
+#include "aidge/operator/Mul.hpp"
+#include "aidge/operator/Producer.hpp"
 #include "aidge/utils/TensorUtils.hpp"
+#include "aidge/utils/Types.h"
 
-#include "aidge/operator/Add.hpp"
 
 namespace Aidge
 {