diff --git a/src/lms/helper.rs b/src/lms/helper.rs
index a483d01..b463a31 100644
--- a/src/lms/helper.rs
+++ b/src/lms/helper.rs
@@ -1,6 +1,6 @@
 use tinyvec::ArrayVec;
 
-use crate::constants::{D_INTR, D_LEAF, MAX_HASH_SIZE};
+use crate::constants::{Node, D_INTR, D_LEAF, MAX_HASH_SIZE};
 use crate::hasher::HashChain;
 use crate::hss::aux::{hss_extract_aux_data, hss_save_aux_data, MutableExpandedAuxData};
 use crate::lm_ots;
@@ -11,8 +11,7 @@ pub fn get_tree_element<H: HashChain>(
     index: usize,
     private_key: &LmsPrivateKey<H>,
     aux_data: &mut Option<MutableExpandedAuxData>,
-) -> ArrayVec<[u8; MAX_HASH_SIZE]> {
-    // TODO/Rework: use "Node" struct as return value?
+) -> Node {
     // Check if we already have the value cached
     if let Some(aux_data) = aux_data {
         if let Some(result) = hss_extract_aux_data::<H>(aux_data, index) {