Add missing include line