fix DI bug.