small cleanup in math