Add missing include from previous commit.