Fixed a bug in the DI file (extension_name) and added unit tests.