Revert "Revert "Fix build error""

This reverts commit 6d03d8da1df661720c0c88382db6f3c934273552.