minor fix: move a print statement