diff --git a/test_device.py b/test_device.py
index acb182b5..cc5e2f05 100755
--- a/test_device.py
+++ b/test_device.py
@@ -36,7 +36,7 @@
 import time
 import unittest
 
-import proto.adb_host_pb2 as adb_host_proto
+import adb_host_pb2 as adb_host_proto
 
 from datetime import datetime