diff --git a/2020/Python/day02.py b/2020/Python/day02.py
index 1f6576a..c286072 100644
--- a/2020/Python/day02.py
+++ b/2020/Python/day02.py
@@ -2,12 +2,11 @@ def part1():
     result = 0
     with open("input2.txt") as f:
         for line in f:
-            (policy, char, passcode) = line.split()
+            (policy, (char,_), passcode) = line.split()
             min_p, max_p = policy.split("-")
-            number_of_char = passcode.count(char[0])
+            number_of_char = passcode.count(char)
             if int(min_p) <= number_of_char <= int(max_p):
                 result += 1
-        
         print(result)
 
 
@@ -15,9 +14,9 @@ def part2():
     result = 0
     with open("input2.txt") as f:
         for line in f:
-            (policy, char, passcode) = line.split()
+            (policy, (char,_), passcode) = line.split()
             min_p, max_p = policy.split("-")
-            if (passcode[int(min_p)-1] == char[0] and passcode[int(max_p)-1] != char[0]) or (passcode[int(min_p)-1] != char[0] and passcode[int(max_p)-1] == char[0]):
+            if (passcode[int(min_p)-1] == char and passcode[int(max_p)-1] != char) or (passcode[int(min_p)-1] != char and passcode[int(max_p)-1] == char):
                 result += 1
         print(result)