diff --git a/api/controller/sdk_unittest.py b/api/controller/sdk_unittest.py
index 7492b26..7774e0a 100644
--- a/api/controller/sdk_unittest.py
+++ b/api/controller/sdk_unittest.py
@@ -10,7 +10,7 @@
 from chromite.lib import cros_test_lib
 
 from chromite.api.controller import sdk as sdk_controller
-from chromite.api.gen import sdk_pb2
+from chromite.api.gen.chromite.api import sdk_pb2
 from chromite.lib import cros_build_lib
 from chromite.service import sdk as sdk_service