Update at_di.module