Oops, I'm a dummy. It was just the print statement that was wrong.