diff --git a/menoh/attribute_completion_and_shape_inference.hpp b/menoh/attribute_completion_and_shape_inference.hpp
index 452afd7..29a6e7b 100644
--- a/menoh/attribute_completion_and_shape_inference.hpp
+++ b/menoh/attribute_completion_and_shape_inference.hpp
@@ -175,7 +175,7 @@ add_variable_to_table(output(0), dtype_of(input(0)),
 else
 
 
-if(node.op_type == "BatchNorm") {
+if(node.op_type == "BatchNormalization") {
     
     
 {
diff --git a/scripts/gen_attribute_completion_and_shape_inference_hpp.py b/scripts/gen_attribute_completion_and_shape_inference_hpp.py
index 18861f5..aa5b677 100644
--- a/scripts/gen_attribute_completion_and_shape_inference_hpp.py
+++ b/scripts/gen_attribute_completion_and_shape_inference_hpp.py
@@ -161,7 +161,7 @@ def main():
 assert(2 <= ndims_of(input(0)));
 '''))
     code_list.append(
-        make_completion_code("BatchNorm", [
+        make_completion_code("BatchNormalization", [
             ("epsilon", "float", "1.e-05f"),
             ("momentum", "float", "0.9f"),
             ("spatial", "int", "1"),