Fixing issue with example output