Fix example in main so that it runs.