Build API: Update imports to new generated proto

BUG=chromium:940582, b:128456793
TEST=run_tests

Change-Id: I34645640bf4fa07b4403f9e2e5e596b7a59dff97
Reviewed-on: https://chromium-review.googlesource.com/1525076
Commit-Ready: ChromeOS CL Exonerator Bot <chromiumos-cl-exonerator@appspot.gserviceaccount.com>
Tested-by: Alex Klein <saklein@chromium.org>
Reviewed-by: Lann Martin <lannm@chromium.org>
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