Fixed missing include (causing compiler warning.)