diff --git a/gcloud/datastore/test_dataset.py b/gcloud/datastore/test_dataset.py
index ccde90f08d5a..5f3e6000055b 100644
--- a/gcloud/datastore/test_dataset.py
+++ b/gcloud/datastore/test_dataset.py
@@ -152,11 +152,9 @@ def test_get_entity_path(self):
         self.assertEqual(result['foo'], 'Foo')
 
     def test_get_entity_odd_nonetype(self):
-        from gcloud.datastore.connection import datastore_pb
         DATASET_ID = 'DATASET'
         KIND = 'Kind'
-        entity_pb = datastore_pb.Entity()
-        connection = _Connection(entity_pb)
+        connection = _Connection()
         dataset = self._makeOne(DATASET_ID, connection)
         with self.assertRaises(ValueError):
             dataset.get_entity([KIND])