From 81f109ea1cad242badbf4e4db8a6073b77eddb7a Mon Sep 17 00:00:00 2001
From: JJ Geewax <jj@geewax.org>
Date: Tue, 22 Apr 2014 12:52:13 -0400
Subject: [PATCH] Fixed #86 - Entity objects from a fetched query have a proper
 Dataset reference.

---
 gcloud/datastore/query.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/gcloud/datastore/query.py b/gcloud/datastore/query.py
index 0e209b01e12b8..e8838de24951f 100644
--- a/gcloud/datastore/query.py
+++ b/gcloud/datastore/query.py
@@ -311,4 +311,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]