diff --git a/android_env/environment.py b/android_env/environment.py
index 8700535..f181556 100644
--- a/android_env/environment.py
+++ b/android_env/environment.py
@@ -22,7 +22,6 @@
 from android_env.components import coordinator as coordinator_lib
 from android_env.proto import adb_pb2
 from android_env.proto import state_pb2
-from android_env.proto import task_pb2
 import dm_env
 import numpy as np