Fix assignment error