diff --git a/etils/ecolab/inspects/nodes.py b/etils/ecolab/inspects/nodes.py
index 675aab8b..9688058e 100644
--- a/etils/ecolab/inspects/nodes.py
+++ b/etils/ecolab/inspects/nodes.py
@@ -31,9 +31,10 @@
 from google.protobuf import descriptor_pool
 from google.protobuf import message
 from google.protobuf import symbol_database
+from google.protobuf.internal import containers
+from google.protobuf.pyext import _message as cpp_containers
 
 # Import both C++ and Python API
-from google.protobuf.internal import containers
 
 _T = TypeVar('_T')