Any reason not to put this in a `visit_Assign` method?