This was broken because multiple packages in the program were named 'main', even one that was imported (by the generated main package). This fixes tests for main packages.
