It might also be that individuals who have these traits self-select into studying math. So we can't easily conclude that studying math leads to the development

It might also be that individuals who have these traits self-select into studying math. So we can't easily conclude that studying math leads to the development of these "highly mathematical habits", because the causation could also flow the other way.