Fixed bug with print statement