Fix figures bug for linear_scale finding