treatment page ui fix