Small assignation fix