diff --git a/tests/test_item_equip.rs b/tests/test_item_equip.rs
index ebc866e..484c161 100644
--- a/tests/test_item_equip.rs
+++ b/tests/test_item_equip.rs
@@ -31,7 +31,7 @@ async fn test_equip_unit_from_equip_menu() {
                 slot: 0,
                 equipped: true,
             }
-        }).await;
+        }).await.unwrap();
 
     entity_gateway.create_item(
         item::NewItemEntity {
@@ -87,7 +87,7 @@ async fn test_equip_unit_from_equip_menu() {
         unknown1: 0,
     })))).await.unwrap().for_each(drop);
 
-    let items = entity_gateway.get_items_by_character(&char1).await;
+    let items = entity_gateway.get_items_by_character(&char1.id).await.unwrap();
     let (unit1, unit2) = (&items[1], &items[2]);
 
     let unit1_equipped = match unit1.location {
@@ -194,7 +194,7 @@ async fn test_unequip_armor_with_units() {
         unknown1: 0,
     })))).await.unwrap().for_each(drop);
 
-    let items = entity_gateway.get_items_by_character(&char1).await;
+    let items = entity_gateway.get_items_by_character(&char1.id).await.unwrap();
     let (armor, unit1, unit2) = (&items[0], &items[1], &items[2]);
 
     let armor_equipped = match armor.location {