Bug fix for code printing.`