diff --git a/quil-rs/src/program/graph.rs b/quil-rs/src/program/graph.rs
index 1e45fa64..45a2e0c9 100644
--- a/quil-rs/src/program/graph.rs
+++ b/quil-rs/src/program/graph.rs
@@ -709,13 +709,14 @@ impl<'a> ScheduledProgram<'a> {
 
 #[cfg(test)]
 mod tests {
-    
-    
-    
+    use super::*;
+    use crate::instruction::Pragma;
+    use crate::program::{MatchedFrames, MemoryAccesses};
 
     #[cfg(feature = "graphviz-dot")]
     mod custom_handler {
         use super::*;
+
         use crate::instruction::PragmaArgument;
         use crate::program::frame::FrameMatchCondition;
         use crate::program::graphviz_dot::tests::build_dot_format_snapshot_test_case;