Fixed missing include and incorrect module statement