diff --git a/src/cleanup_test.rs b/src/cleanup_test.rs
index 367c7e3..c68bb9c 100644
--- a/src/cleanup_test.rs
+++ b/src/cleanup_test.rs
@@ -1,8 +1,16 @@
-
-
 #[cfg(test)]
 mod cleanup_tests {
-    use super::*;
+    use bollard::Docker;
+    use std::process::Command;
+    use crate::{
+        executor::{
+            docker,
+        },
+        runtime::{
+            emulation::{self, EmulationRuntime},
+        },
+        cleanup_on_exit,
+    };
 
     #[tokio::test]
     async fn test_docker_container_cleanup() {