Fixed test that was broken when CassandraDataStore interface was refactored.