Upgraded to di 2.0.1