diff --git a/src/attention.rs b/src/attention.rs
index b72079b..e394785 100644
--- a/src/attention.rs
+++ b/src/attention.rs
@@ -42,10 +42,7 @@ impl From<Attention> for Element {
 
 #[cfg(test)]
 mod tests {
-    use try_from::TryFrom;
-    use minidom::Element;
-    use error::Error;
-    use super::Attention;
+    use super::*;
 
     #[test]
     fn test_simple() {