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

This file does not need to be changed, since there are no statements executed except for the warning.