Fix cqlsh tests