diff --git a/tests/test_basic.rs b/tests/test_basic.rs
index dfdc5f8..bbc9f16 100644
--- a/tests/test_basic.rs
+++ b/tests/test_basic.rs
@@ -1,4 +1,4 @@
-use chainerror::ResultTrait;
+use chainerror::Context;
 
 #[test]
 fn test_basic() {
diff --git a/tests/test_iter.rs b/tests/test_iter.rs
index 3a464b1..9023078 100644
--- a/tests/test_iter.rs
+++ b/tests/test_iter.rs
@@ -1,4 +1,4 @@
-use chainerror::ResultTrait;
+use chainerror::Context;
 use std::error::Error;
 use std::io;