X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=bindings%2Fpython%2Fbt2%2Fbt2%2Fstream_class.py;fp=bindings%2Fpython%2Fbt2%2Fbt2%2Fstream_class.py;h=54b5b38bba0587bf1f83e5ff9f3f44e549a32043;hp=684b952b1f970c5b400ec83c0d7cd2f788f6cdb7;hb=2c6f8520ad68e1ca0c9554d705c39790e7d3ef5f;hpb=0fd756a43bce605875565a14c1ed1b070fa3ad94

diff --git a/bindings/python/bt2/bt2/stream_class.py b/bindings/python/bt2/bt2/stream_class.py
index 684b952b..54b5b38b 100644
--- a/bindings/python/bt2/bt2/stream_class.py
+++ b/bindings/python/bt2/bt2/stream_class.py
@@ -22,12 +22,13 @@
 
 from bt2 import native_bt, object, utils
 import bt2.field_class
+import bt2.event_class
 import collections.abc
 import bt2.stream
 import bt2
 
 
-class StreamClass(object._SharedObject, collections.abc.Mapping):
+class _StreamClass(object._SharedObject, collections.abc.Mapping):
     _get_ref = staticmethod(native_bt.stream_class_get_ref)
     _put_ref = staticmethod(native_bt.stream_class_put_ref)
 
@@ -38,7 +39,7 @@ class StreamClass(object._SharedObject, collections.abc.Mapping):
         if ec_ptr is None:
             raise KeyError(key)
 
-        return bt2.EventClass._create_from_ptr_and_get_ref(ec_ptr)
+        return bt2.event_class._EventClass._create_from_ptr_and_get_ref(ec_ptr)
 
     def __len__(self):
         count = native_bt.stream_class_get_event_class_count(self._ptr)
@@ -70,7 +71,7 @@ class StreamClass(object._SharedObject, collections.abc.Mapping):
             utils._check_uint64(id)
             ec_ptr = native_bt.event_class_create_with_id(self._ptr, id)
 
-        event_class = bt2.event_class.EventClass._create_from_ptr(ec_ptr)
+        event_class = bt2.event_class._EventClass._create_from_ptr(ec_ptr)
 
         if name is not None:
             event_class._name = name
@@ -94,7 +95,7 @@ class StreamClass(object._SharedObject, collections.abc.Mapping):
         tc_ptr = native_bt.stream_class_borrow_trace_class_const(self._ptr)
 
         if tc_ptr is not None:
-            return bt2.TraceClass._create_from_ptr_and_get_ref(tc_ptr)
+            return bt2._TraceClass._create_from_ptr_and_get_ref(tc_ptr)
 
     @property
     def name(self):