Fix tests

hectorhdzg · hectorhdzg · commit a714d204bd9f · 2019-10-15T15:04:04.000-07:00
diff --git a/opentelemetry-sdk/tests/trace/test_trace.py b/opentelemetry-sdk/tests/trace/test_trace.py
@@ -291,13 +291,13 @@ def test_start_span(self):
         # default status
         self.assertTrue(span.status.is_ok)
         self.assertEqual(
-            span.status.canonical_code, trace_api.StatusCanonicalCode.OK
+            span.status.canonical_code, trace_api.status.StatusCanonicalCode.OK
         )
         self.assertIs(span.status.description, None)
 
         # status
-        new_status = trace_api.Status(
-            trace_api.StatusCanonicalCode.CANCELLED, "Test description"
+        new_status = trace_api.status.Status(
+            trace_api.status.StatusCanonicalCode.CANCELLED, "Test description"
         )
         span.set_status(new_status)
         self.assertEqual(span.status, new_status)
