Modification to staleness code