Add a missing include