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

In this case, the file does not need modifications, since statements are already contained in an `if __name__ == "__main__"` conditional statement.