Fix for the automated tests that don't run the demos as __main__.