bug fix in di