Fixed missing include statement