(fix) Update print statements