From 62208d5f512f86e2b3a4a01b418f99597a315b3b Mon Sep 17 00:00:00 2001
From: tserg <8017125+tserg@users.noreply.github.com>
Date: Sun, 7 Jan 2024 11:48:43 +0800
Subject: [PATCH] fix tests

---
 tests/unit/compiler/asm/test_asm_optimizer.py | 2 +-
 tests/unit/compiler/test_source_map.py        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/unit/compiler/asm/test_asm_optimizer.py b/tests/unit/compiler/asm/test_asm_optimizer.py
index 44b823757c..b2851e908a 100644
--- a/tests/unit/compiler/asm/test_asm_optimizer.py
+++ b/tests/unit/compiler/asm/test_asm_optimizer.py
@@ -58,7 +58,7 @@ def ctor_only():
 
 @internal
 def runtime_only():
-    for i in range(10):
+    for i: uint256 in range(10):
         self.s += 1
 
 @external
diff --git a/tests/unit/compiler/test_source_map.py b/tests/unit/compiler/test_source_map.py
index c9a152b09c..5b478dd2aa 100644
--- a/tests/unit/compiler/test_source_map.py
+++ b/tests/unit/compiler/test_source_map.py
@@ -6,7 +6,7 @@
 @internal
 def _baz(a: int128) -> int128:
     b: int128 = a
-    for i in range(2, 5):
+    for i: int128 in range(2, 5):
         b *=  i
         if b > 31337:
             break