Fix mock cassandra client return value