diff --git a/ui_automator/ui_automator.py b/ui_automator/ui_automator.py
index f78d812..74698f0 100644
--- a/ui_automator/ui_automator.py
+++ b/ui_automator/ui_automator.py
@@ -18,6 +18,7 @@
 automated UI operations on Android phones.
 """
 import collections
+from collections.abc import Callable
 from concurrent import futures
 import enum
 import functools
@@ -25,7 +26,7 @@
 import os
 import re
 import time
-from typing import Any, Callable
+from typing import Any
 import unittest
 
 from absl import app