diff --git a/gcloud/datastore/query.py b/gcloud/datastore/query.py index 46c34c1e0933..9b89c55ca4bc 100644 --- a/gcloud/datastore/query.py +++ b/gcloud/datastore/query.py @@ -244,4 +244,5 @@ def fetch(self, limit=None): entity_pbs = self.dataset().connection().run_query( query_pb=clone.to_protobuf(), dataset_id=self.dataset().id()) - return [Entity.from_protobuf(entity) for entity in entity_pbs] + return [Entity.from_protobuf(entity, dataset=self.dataset()) + for entity in entity_pbs]