diff --git a/replace.py b/replace.py
index 508c3f33f32c42600dc9aa25c21f62bbaf119dd3..2e8c8c5960189d02f0ab35eec90789c6f519aa45 100644
--- a/replace.py
+++ b/replace.py
@@ -9,7 +9,7 @@ def replace_type_hints(file_path):
     # Decode the file data with error handling
     file_data = file_data.decode("utf-8", errors="ignore")
 
-    # Regular expression pattern to find 'Optional[dict[int, int]]' and replace with 'Optional[dict[int, int]]'
+    # Regular expression pattern to find 'dict[Type1, Type2] | None' and replace with 'Optional[dict[Type1, Type2]]'
     file_data = re.sub(
         r"dict\[(\w+), (\w+)\]\s*\|\s*None", r"Optional[dict[\1, \2]]", file_data
     )