Remove print statement
Try to fix bug