Fix build

(cherry picked from commit 787aec8f8eef497e597d66d6acd9138f800f199a)