Fixed update test to pass in the correct keyspace