Add another missing include