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/api/controller/test_archive_unittest.py b/api/controller/test_archive_unittest.py
index 9c73b8c..9c2f2af 100644
--- a/api/controller/test_archive_unittest.py
+++ b/api/controller/test_archive_unittest.py
@@ -10,7 +10,7 @@
 import os
 
 from chromite.api.controller import test_archive as test_archive_controller
-from chromite.api.gen import test_archive_pb2
+from chromite.api.gen.chromite.api import test_archive_pb2
 from chromite.lib import cros_build_lib
 from chromite.lib import cros_test_lib
 from chromite.lib import osutils