Added option to have ability to return module extension