tests: Replace umodule with module everywhere.

This work was funded through GitHub Sponsors.

Signed-off-by: Jim Mussared <jim.mussared@gmail.com>

1 files changed, 5 insertions, 6 deletions

diff --git a/tests/thread/stress_create.py b/tests/thread/stress_create.py
index 877424cdf..98f272f5c 100644
--- a/tests/thread/stress_create.py
+++ b/tests/thread/stress_create.py
@@ -1,13 +1,12 @@
 # stress test for creating many threads
 
-try:
-    import utime
-
-    sleep_ms = utime.sleep_ms
-except ImportError:
-    import time
+import time
 
+if hasattr(time, "sleep_ms"):
+    sleep_ms = time.sleep_ms
+else:
     sleep_ms = lambda t: time.sleep(t / 1000)
+
 import _thread