diff --git a/src/py/flwr_tool/protoc_test.py b/src/py/flwr_tool/protoc_test.py
index 607d808c8497..2d48582eb441 100644
--- a/src/py/flwr_tool/protoc_test.py
+++ b/src/py/flwr_tool/protoc_test.py
@@ -28,4 +28,4 @@ def test_directories() -> None:
 
 def test_proto_file_count() -> None:
     """Test if the correct number of proto files were captured by the glob."""
-    assert len(PROTO_FILES) == 6
+    assert len(PROTO_FILES) == 7