From 29246a483e26eaab4d645a7864720937cdce6d7c Mon Sep 17 00:00:00 2001
From: huntc <huntchr@gmail.com>
Date: Thu, 24 Aug 2023 12:35:01 +1000
Subject: [PATCH] Type corrections

---
 akka-persistence-rs/src/lib.rs | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/akka-persistence-rs/src/lib.rs b/akka-persistence-rs/src/lib.rs
index d06755c..35f5cb4 100644
--- a/akka-persistence-rs/src/lib.rs
+++ b/akka-persistence-rs/src/lib.rs
@@ -118,8 +118,6 @@ pub trait WithOffset {
 
 #[cfg(test)]
 mod tests {
-    use smol_str::SmolStr;
-
     use super::*;
 
     #[test]
@@ -133,8 +131,8 @@ mod tests {
     fn test_slice_for_persistence_id() {
         assert_eq!(
             PersistenceId::new(
-                SmolStr::from("some-entity-type"),
-                SmolStr::from("some-entity-id")
+                EntityType::from("some-entity-type"),
+                EntityId::from("some-entity-id")
             )
             .slice(),
             451