Fix build

(cherry picked from commit 28d43cf3c0b6f9a7a19630fe8d42f72a56f81b0d)