Fix build: missing changes from previous commit.