added missing include for entity file