From a3c9fb7b55eb915f5e98dfe9f8e847f58fab2c20 Mon Sep 17 00:00:00 2001
From: NewSoupVi <57900059+NewSoupVi@users.noreply.github.com>
Date: Fri, 7 Jun 2024 22:10:36 +0200
Subject: [PATCH] Improve typing

---
 worlds/yachtdice/__init__.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/worlds/yachtdice/__init__.py b/worlds/yachtdice/__init__.py
index 43e8244a69..cf541c9f59 100644
--- a/worlds/yachtdice/__init__.py
+++ b/worlds/yachtdice/__init__.py
@@ -1,7 +1,7 @@
 import math
 import logging
 
-from BaseClasses import Region, Entrance, Item, Tutorial
+from BaseClasses import Region, Entrance, Item, Tutorial, CollectionState
 from .Items import YachtDiceItem, item_table, item_groups
 from .Locations import YachtDiceLocation, all_locations, ini_locations
 from .Options import YachtDiceOptions
@@ -397,14 +397,14 @@ class YachtDiceWorld(World):
         item = YachtDiceItem(name, item_data.classification, item_data.code, self.player)
         return item
 
-    def collect(self, state, item: Item) -> bool:
+    def collect(self, state: CollectionState, item: Item) -> bool:
         change = super().collect(state, item)
         if change and item.advancement:
             state.prog_items[self.player]["state_is_fresh"] = 0
 
         return change
 
-    def remove(self, state, item: Item) -> bool:
+    def remove(self, state: CollectionState, item: Item) -> bool:
         change = super().remove(state, item)
         if change and item.advancement:
             state.prog_items[self.player]["state_is_fresh"] = 0