fix type

EspressoSystems · Mar 18, 2024 · b8ec0be · b8ec0be
1 parent e6099e7
commit b8ec0be
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/traits/storage.rs b/src/traits/storage.rs
@@ -6,13 +6,16 @@
 use anyhow::Result;
 use async_trait::async_trait;
 
-use crate::data::{DAProposal, VidDisperse};
+use crate::{
+    data::{DAProposal, VidDisperse},
+    message::Proposal,
+};
 
 use super::node_implementation::NodeType;
 
 /// Abstraction for storing a variety of consensus payload datum.
 #[async_trait]
 pub trait Storage<TYPES: NodeType>: Send + Sync + Clone {
-    async fn append_vid(&self, proposal: &DAProposal<TYPES>) -> Result<()>;
-    async fn append_da(&self, proposal: &VidDisperse<TYPES>) -> Result<()>;
+    async fn append_vid(&self, proposal: &Proposal<TYPES, DAProposal<TYPES>>) -> Result<()>;
+    async fn append_da(&self, proposal: &Proposal<TYPES, VidDisperse<TYPES>>) -> Result<()>;
 }