add missing import due to cherry-pick