datastore: Fixing regression test to support the Key type.