Fix small bug restricting domain of treedepth correlation plots