diff --git a/leetcode/202.py b/leetcode/202.py
index e747540..6e8522f 100644
--- a/leetcode/202.py
+++ b/leetcode/202.py
@@ -8,7 +8,7 @@ class Solution:
         str_n = str(n)
         total = 0
         for number in str_n:
-            total += number * number
+            total += str(number) * str(number)
 
         if total == 1:
             return True