Fix treatment effect