Added repartition. Carefull = is not copy. Maybe need to fix errors like this in the future