Trump tried to have only citizens and legal residents count for the 2020 census. Some Court struck that down.
If I remember correctly, Roberts caused the problem. He wanted “lawful justification” for asking the citizenship question on the census.