diff --git a/device.mk b/device.mk
index 678daf1..faa0dd8 100644
--- a/device.mk
+++ b/device.mk
@@ -1045,7 +1045,7 @@ PRODUCT_PROPERTY_OVERRIDES += persist.vendor.enable.thermal.genl=true
 # EdgeTPU
 include device/google/gs-common/edgetpu/edgetpu.mk
 # Config variables for TPU chip on device.
-$(call soong_config_set,edgetpu_config,chip,rio)
+$(call soong_config_set,edgetpu_config,chip,rio_pro)
 # TPU firmware
 PRODUCT_PACKAGES += edgetpu-rio.fw