Fix #50 while keeping the full example intact.

Thanks, @therealadam!