Add missing include directive