Run tests with both test and main classes - might work this way.