diff --git a/scripts/build_api_unittest.py b/scripts/build_api_unittest.py
index a86d488..aad9a08 100644
--- a/scripts/build_api_unittest.py
+++ b/scripts/build_api_unittest.py
@@ -7,7 +7,7 @@
 
 from __future__ import print_function
 
-from chromite.api.gen import build_api_test_pb2
+from chromite.api.gen.chromite.api import build_api_test_pb2
 from chromite.lib import cros_build_lib
 from chromite.lib import cros_test_lib
 from chromite.scripts import build_api