Fix build from previous commit