diff --git a/src/drisl.rs b/src/drisl.rs
index e90d03d..d55f12a 100644
--- a/src/drisl.rs
+++ b/src/drisl.rs
@@ -45,6 +45,7 @@ mod tests {
     #[derive(Serialize, Deserialize, Clone, Debug, PartialEq)]
     struct Struct<'a> {
         tuple_struct: TupleStruct,
+        unit_struct: UnitStruct,
         tuple: (String, f32, f64),
         map: BTreeMap<String, String>,
         #[serde(with = "serde_bytes")]
@@ -75,6 +76,7 @@ mod tests {
         let array = vec!["one".to_string(), "two".to_string(), "three".to_string()];
         let data = Struct {
             tuple_struct,
+            unit_struct: UnitStruct,
             tuple,
             map,
             bytes,
diff --git a/tests/enum.rs b/tests/enum.rs
index 4605c9b..8b2bdad 100644
--- a/tests/enum.rs
+++ b/tests/enum.rs
@@ -85,11 +85,3 @@ fn test_variable_length_array_error() {
     let err = value.unwrap_err();
     assert!(matches!(err, DecodeError::IndefiniteSize), "{err:?}");
 }
-
-#[derive(Serialize, Deserialize, PartialEq, Debug)]
-enum Bar {
-    Empty,
-    Number(i32),
-    Flag(String, bool),
-    Point { x: i32, y: i32 },
-}