I'd have changed this in the consistency level inside of the get_consistency_level function, myself.