Made a quick change to the main module