Added a reference inside the DI file.