fix DI