Fix build after latest commit