Fixed missing include for unit tests.