Improves Cassandra testability

I needed this to work on #268.