diff --git a/concrete/numpy/tracing.py b/concrete/numpy/tracing.py
index 24e19f74b..c62068f7f 100644
--- a/concrete/numpy/tracing.py
+++ b/concrete/numpy/tracing.py
@@ -271,7 +271,7 @@ class NPTracer(BaseTracer):
         numpy.sinh,
         numpy.spacing,
         numpy.sqrt,
-        # numpy.square,
+        numpy.square,
         # numpy.subtract,
         numpy.tan,
         numpy.tanh,