diff --git a/src/main.rs b/src/main.rs
index 1642263..ff98b9e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,4 @@
-use doublets::{data::LinkType, mem, unit, Doublets, Error};
+use doublets::{data::LinkType, mem, unit, Doublets, Error, LinkType};
 use doublets::{DoubletsExt, Links};
 use tap::Pipe;
 
@@ -67,7 +67,8 @@ fn nand(a: bool, b: bool) -> bool {
     !(a && b)
 }
 
-fn get_link_by_id(store: &mut unit::Store<usize, _>, id: usize) -> Result<Link<usize>, Error<usize>> {
+fn get_link_by_id<T: LinkType>(store: &mut impl Doublets<T>, id: usize) -> Result<Link<T>, Error<usize>> 
+{
     // `any` constant denotes any link
     let any = store.constants().any;
     let mut link_result = Err(Error::new("Link not found"));