Fix cassandra test