fix test references to main module