X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=src%2Fbindings%2Fpython%2Fbt2%2Fbt2%2Fstream_class.py;fp=src%2Fbindings%2Fpython%2Fbt2%2Fbt2%2Fstream_class.py;h=cda21864b5c409b515a611275debcc02f06820a3;hp=08186333f1257f7591edcfe367d09f062c47b3a4;hb=5290f876138c2b71026bd2bbed78077b68cf961a;hpb=68e606effd31ee0b35d243f0c406c8af0f4a7b17

diff --git a/src/bindings/python/bt2/bt2/stream_class.py b/src/bindings/python/bt2/bt2/stream_class.py
index 08186333..cda21864 100644
--- a/src/bindings/python/bt2/bt2/stream_class.py
+++ b/src/bindings/python/bt2/bt2/stream_class.py
@@ -23,6 +23,7 @@
 from bt2 import native_bt, object, utils
 import bt2.field_class
 import bt2.event_class
+import bt2.trace_class
 import collections.abc
 import bt2.stream
 import bt2
@@ -107,7 +108,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.trace_class._TraceClass._create_from_ptr_and_get_ref(tc_ptr)
 
     @property
     def name(self):