Fix keyspace_test