diff --git a/src/generator.rs b/src/generator.rs
index 16ba2de..972d0a0 100644
--- a/src/generator.rs
+++ b/src/generator.rs
@@ -209,7 +209,7 @@ impl CodeGen for Block {
 
                 use std::marker::PhantomData;
 
-                use super::RegisterInterface;
+                use super::#parent;
 
                 #(#code_children)*
 
@@ -217,7 +217,7 @@ impl CodeGen for Block {
 
                 pub struct #upper_camel_name<'a> {
                     mem_ptr: *mut u32,
-                    _marker: PhantomData<&'a mut RegisterInterface>,
+                    _marker: PhantomData<&'a mut #parent_struct>,
                 }
 
                 impl #upper_camel_name<'_> {