Fix test, remove debug statements in tests