bugfix in df.treatment