Add missing include statement