fix datastore tests