attempt to fix DI