Fix files that do work on import instead of only if __name__ == __main__