Fix the tests and remove a pring statement