add wrapper in the module