Fixed main module for test runs