Fix printing in code that was just added