Remove unwanted code from the module